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])