问题描述:
input_x = Tensor(np.ones(shape=[3,1,1, 1, 1]), mindspore.float32) squeeze = ops.Squeeze(3).Squeeze(3) output = squeeze(input_x) print(output.shape)
解答:
可以使用Tensor自带squeeze方法,也就是input_x.squeeze(3).squeeze(3).squeeze(),因为squeeze返回的任然是一个Tensor,可以继续使用squeeze方法,您这里实例化ops.Squeeze之后在.Squeeze(),这样是会报错的,ops.Squeeze没有Squeeze这个方法。
有关ops.Squeeze使用可以查看官网https://www.mindspore.cn/docs/api/zh-CN/master/api_python/ops/mindspore.ops.Squeeze.html#mindspore.ops.Squeeze,这里也有Tensor自带的一些属性您也可以看一下https://www.mindspore.cn/docs/api/zh-CN/master/api_python/mindspore/mindspore.Tensor.html#mindspore.Tensor,还有问题,也欢迎继续提问



