- Tensor.reshape(*shape) Tensor ¶
Returns a tensor with the same data and number of elements as
selfbut with the specified shape. This method returns a view if
shapeis compatible with the current shape. See
torch.Tensor.view()on when it is possible to return a view.
shape (tuple of ints or int...) – the desired shape