快捷方式

torch.unbind

torch.unbind(input, dim=0) seq

移除张量维度。

返回沿着给定维度的所有切片,已移除该维度。

参数
  • input (Tensor) – 要解绑的张量

  • dim (int) – 要移除的维度

示例

>>> torch.unbind(torch.tensor([[1, 2, 3],
>>>                            [4, 5, 6],
>>>                            [7, 8, 9]]))
(tensor([1, 2, 3]), tensor([4, 5, 6]), tensor([7, 8, 9]))

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源