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