torch.reshape

torch.reshape(input, shape) → Tensor

返回一個張量，該張量具有與 input 相同的資料和元素數量，但具有指定的形狀。如果可能，返回的張量將會是 input 的一個視圖 (view)。 否則，它將會是一個副本。連續的輸入和具有兼容步幅的輸入可以被重塑而無需複製，但你不應該依賴複製或視圖的行為。

請參閱 torch.Tensor.view() 以了解何時可以返回一個視圖。

可以將單個維度設為 -1，在這種情況下，該維度會根據剩餘的維度和 input 中的元素數量來推斷。

參數

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

• shape (int 的 tuple) – 新的形狀

範例

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