torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

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

См. torch.reshape()

Параметры

shape (кортеж целых чисел или целое число...) – желаемая форма