torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

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

参见 torch.reshape()

参数

shape (元组 of ints 或 int...) – 指定所需的形状