Python PyTorch unbind用法及代碼示例

本文簡要介紹python語言中 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]))