torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

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

См. torch.reshape()

Параметры:

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