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)