mindspore.Tensor.reshape
- Tensor.reshape(*shape)[source]
Give a new shape to a tensor without changing its data.
- Parameters
shape (Union[int, tuple(int), list(int)]) – The new shape should be compatible with the original shape. If an integer, then the result will be a 1-D tensor of that length. One shape dimension can be -1. In this case, the value is inferred from the length of the tensor and remaining dimensions.
- Returns
Tensor, with new specified shape.
- Raises
TypeError – If new shape is not integer, list or tuple.
ValueError – If new shape is not compatible with the original shape.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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]]