torch.Tensor.reshape

Tensor.reshape(*shape) → 张量

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

参见 torch.reshape()

Parameters

形状 (元组 的 整数 或 整数...) – 所需的形状