torch.reshape

torch.reshape(input, shape) → 张量

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

参见 torch.Tensor.view() 以了解何时可以返回视图。

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

Parameters

• 输入 (张量) – 要重塑的张量

• 形状 (元组 的 整数) – 新的形状

示例：

>>> 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])