快捷方式

torch.reshape

torch.reshape(input, shape) Tensor

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

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

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

参数
  • input (Tensor) – 要重塑的张量

  • shape (tuple of int) – 新形状

示例

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源