mindspore.ops.Squeeze
- class mindspore.ops.Squeeze(axis=())[源代码]
返回删除指定 axis 中大小为1的维度后的Tensor。
更多参考详见
mindspore.ops.squeeze()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> input_x = Tensor(np.ones(shape=[3, 2, 1]), mindspore.float32) >>> squeeze = ops.Squeeze(2) >>> output = squeeze(input_x) >>> print(output) [[1. 1.] [1. 1.] [1. 1.]]