torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

返回一个具有与 self 相同数据和元素数量的张量，但具有指定的形状。此方法如果 shape 与当前形状兼容，则返回一个视图。有关何时可以返回视图的信息，请参阅 torch.Tensor.view() 。

查看 torch.reshape()

参数:

形状（整数元组或整数...）- 所需的形状