torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

Возвращает тензор с теми же данными и количеством элементов, что и self, но с указанной формой. Этот метод возвращает представление, если shape совместима с текущей формой. См. torch.Tensor.view(), чтобы узнать, когда можно вернуть представление.

См. torch.reshape()

Параметры:

shape (tuple целых чисел или int...) – требуемая форма