torch.reshape

torch.reshape(input, shape) → Tensor

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

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

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

Параметры:

• input (Tensor) – тензор, подлежащий преобразованию в форму
• shape (кортеж python: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])