mindspore.ops.squeeze

mindspore.ops.squeeze(input, axis=None)[源代码]

删除输入tensor中长度为1的轴。

说明

请注意，在动态图模式下，输出tensor将与输入tensor共享数据，并且没有tensor数据复制过程。
维度索引从0开始，并且必须在 [-input.ndim, input.ndim) 范围内。
在GE模式下，只支持对input shape中大小为1的维度进行压缩。

警告

这是一个实验性API，后续可能修改或删除。

参数：

input (Tensor) - 输入tensor。
axis (Union[int, tuple(int), list(int)]) - 待删除的轴，默认 None 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.ops.ones(shape=[3, 2, 1])
>>> output = mindspore.ops.squeeze(input)
>>> print(output)
[[1. 1.]
 [1. 1.]
 [1. 1.]]