torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

返回一個張量，它具有與 self 相同的資料和元素數量，但具有指定的形狀。如果 shape 與當前形狀相容，則此方法返回一個檢視。請參閱 torch.Tensor.view() 關於何時可以返回檢視。

請參閱 torch.reshape()

引數

shape (元組 of int 或 int...) – 期望的形狀