torch.reshape — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.reshape. torch.reshape(input, shape) → Tensor. Returns a tensor with the same data and number of elements as input , but with the specified shape. When possible, the returned tensor will be a view of input. Otherwise, it will be a copy. Contiguous inputs and inputs with compatible strides can be reshaped without copying, but you should ...