mindspore.Tensor.reshape

mindspore.Tensor.reshape(*shape)

不改变数据的情况下，将Tensor的shape改为输入的新shape。

参数：

• shape (Union[int, tuple(int), list(int)]) - 新的shape应与原来的shape兼容。如果参数值为整数，则结果是该长度的一维数组。shape的维度可以为-1。在这种情况下，将根据数组的长度和剩下的维度计算出该值。

返回：

Tensor，具有新shape的Tensor。

异常：

• TypeError - 新shape不是整数、列表或元组。

• ValueError - 新shape与原来Tensor的shape不兼容。

支持平台：

Ascend GPU CPU

样例：

>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> x = Tensor([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]], dtype=mstype.float32)
>>> output = x.reshape((3, 2))
>>> print(output)
[[-0.1  0.3]
[ 3.6  0.4]
[ 0.5 -3.2]]