torch.reshape

torch.reshape(input, shape) → Tensor

Возвращает тензор с теми же данными и количеством элементов, что и input, но с указанной формой. Когда это возможно, возвращаемый тензор будет представлением input. В противном случае будет создана копия. Непрерывные входные данные и входные данные с совместимыми шагами могут быть преобразованы в другую форму без копирования, однако не следует полагаться на то, будет ли создана копия или представление.

См. torch.Tensor.view(), чтобы узнать, когда возможно вернуть представление.

Одно измерение может быть равно -1; в этом случае оно выводится из оставшихся измерений и количества элементов в input.

Параметры:

• input (Tensor) – тензор, форму которого нужно изменить
• shape (tuple of int) – новая форма

Пример:

>>> a = torch.arange(4.)
>>> torch.reshape(a, (2, 2))
tensor([[ 0.,  1.],
        [ 2.,  3.]])
>>> b = torch.tensor([[0, 1], [2, 3]])
>>> torch.reshape(b, (-1,))
tensor([ 0,  1,  2,  3])