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