torch.Tensor.reshape

Tensor.reshape(*shape) → Tensor

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

请参阅 torch.reshape()

参数

shape (元组 的 整数 或 整数...) – 期望的形状