Python PyTorch reshape用法及代码示例

本文简要介绍python语言中 torch.reshape 的用法。

用法:

torch.reshape(input, shape) → Tensor

参数：

• input(Tensor) -要重塑的张量

• shape(python的元组：ints) -新形状

返回与 input 具有相同数据和元素数量的张量，但具有指定的形状。如果可能，返回的张量将是 input 的视图。否则，它将是副本。可以在不复制的情况下重新调整连续输入和具有兼容步幅的输入，但您不应依赖于复制与查看行为。

当可以返回视图时，请参阅 torch.Tensor.view() 。

单个维度可能是 -1，在这种情况下，它是从剩余维度和 input 中的元素数推断出来的。

例子：

>>> a = torch.arange(4.)
>>> torch.reshape(a, (2, 2))
tensor([[ 0.,  1.],
        [ 2.,  3.]])
>>> b = torch.tensor([[0, 1], [2, 3]])
>>> torch.reshape(b, (-1,))
tensor([ 0,  1,  2,  3])