Python PyTorch reshape用法及代碼示例

本文簡要介紹python語言中 torch.reshape 的用法。

用法:

torch.reshape(input, shape) → Tensor

參數：

• input(Tensor) -要重塑的張量

• shape(python的元組：ints) -新形狀

返回與 input 具有相同數據和元素數量的張量，但具有指定的形狀。如果可能，返回的張量將是 input 的視圖。否則，它將是副本。可以在不複製的情況下重新調整連續輸入和具有兼容步幅的輸入，但您不應依賴於複製與查看行為。

當可以返回視圖時，請參閱 torch.Tensor.view() 。

單個維度可能是 -1，在這種情況下，它是從剩餘維度和 input 中的元素數推斷出來的。

例子：

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