快捷方式

torch.flatten

torch.flatten(input, start_dim=0, end_dim=-1) Tensor

通过将 input 重塑为一维张量来展平它。如果传入 start_dimend_dim,则只展平从 start_dim 开始到 end_dim 结束的维度。input 中元素的顺序保持不变。

与 NumPy 的 flatten 不同,后者总是复制输入数据,此函数可能会返回原始对象、视图或副本。如果未展平任何维度,则返回原始对象 input。否则,如果 input 可以被视为展平后的形状,则返回该视图。最后,只有当输入不能被视为展平后的形状时,才会复制输入的数据。关于何时返回视图的详细信息,请参阅 torch.Tensor.view()

注意

展平零维张量将返回一个一维视图。

参数
  • input (Tensor) – 输入张量。

  • start_dim (int) – 要展平的起始维度

  • end_dim (int) – 要展平的结束维度

示例

>>> t = torch.tensor([[[1, 2],
...                    [3, 4]],
...                   [[5, 6],
...                    [7, 8]]])
>>> torch.flatten(t)
tensor([1, 2, 3, 4, 5, 6, 7, 8])
>>> torch.flatten(t, start_dim=1)
tensor([[1, 2, 3, 4],
        [5, 6, 7, 8]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深度教程

查看教程

资源

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

查看资源