pytorch每日一学46(torch.unbind())返回给定维度所有切片的元组

第46个方法

torch.unbind(input, dim=0) → seq

此方法就是将我们的input从dim进行切片，并返回切片的结果，返回的结果里面没有dim这个维度。

例如维度为0的时候，如下所示：

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

就是从0维将其分解掉，如果为1的时候，就是按列将其分解：

分解的元组数量就是指定dim维中的元素数量。