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的时候,就是按列将其分解:
pytorch每日一学46(torch.unbind())返回给定维度所有切片的元组_第1张图片
分解的元组数量就是指定dim维中的元素数量。
pytorch每日一学46(torch.unbind())返回给定维度所有切片的元组_第2张图片

你可能感兴趣的:(pytorch每日一学,深度学习,pytorch,神经网络,机器学习,数据挖掘)