def squeeze(x, axis):
'''Remove a 1-dimension from the tensor at index "axis".
'''
shape = list(x.shape)
shape.pop(axis)
return T.reshape(x, tuple(shape))
微信
公众号
扫码关注公众号