mindspore.ops.Squeeze

查看源文件
class mindspore.ops.Squeeze(axis=())[源代码]

返回删除指定 axis 中大小为1的维度后的Tensor。

更多参考详见 mindspore.ops.squeeze()

警告

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

参数:
  • axis (Union[int, tuple(int)]) - 指定待删除shape的维度索引,它会删除给定 axis 参数中所有大小为1的维度。如果指定了维度索引,其数据类型必须为int32或int64。默认值: ()

输入:
  • input (Tensor) - 用于计算Squeeze的输入Tensor,shape为 \((x_1, x_2, ..., x_R)\)

输出:

Tensor,shape为 \((x_1, x_2, ..., x_S)\)

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input = Tensor(np.ones(shape=[3, 2, 1]), mindspore.float32)
>>> squeeze = ops.Squeeze(2)
>>> output = squeeze(input)
>>> print(output)
[[1. 1.]
 [1. 1.]
 [1. 1.]]