mindspore.ops.squeeze

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

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

警告

2.9.0（不含）之后版本非兼容性变更：axis 将改为 dim；axis=None 将改为 dim=()，并继续支持 tuple 形式的维度输入。

说明

  • 请注意，在动态图模式下，输出tensor将与输入tensor共享数据，并且没有tensor数据复制过程。

  • 维度索引从0开始，并且必须在 [-input.ndim, input.ndim) 范围内。

参数：

  • 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.]]