mindspore.Tensor.squeeze

mindspore.Tensor.squeeze(axis=None)[源代码]

从Tensor中删除shape为1的维度。

参数:
  • axis (Union[None, int, list(int), tuple(int)], 可选) - 选择shape中长度为1的条目的子集。如果选择shape条目长度大于1的轴,则报错。默认值为None。

返回:

Tensor,删除了长度为1的维度的全部子集或一个子集。

异常:
  • TypeError - 输入的参数类型有误。

  • ValueError - 指定维度的shape大于1。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,2), dtype=np.float32))
>>> print(x)
[[[1. 1.]
[1. 1.]]]
>>> print(x.shape)
(1, 2, 2)
>>> y = x.squeeze()
>>> print(y)
[[1. 1.]
[1. 1.]]
>>> print(y.shape)
(2, 2)
>>> y = x.squeeze(axis=0)
>>> print(y)
[[1. 1.]
[1. 1.]]
>>> print(y.shape)
(2, 2)