Python PyTorch combinations用法及代碼示例

本文簡要介紹python語言中 torch.combinations 的用法。

用法:

torch.combinations(input, r=2, with_replacement=False) → seq

參數：

• input(Tensor) -一維向量。

• r(int,可選的) -要組合的元素數量

• with_replacement(布爾值,可選的) -是否允許組合重複

返回：

一個張量相當於把所有輸入的張量轉成列表，對這些列表做itertools.combinations或者itertools.combinations_with_replacement，最後把得到的列表轉成張量。

返回類型：

Tensor

計算給定張量的長度 r 的組合。當 with_replacement 設置為 False 和 itertools.combinations_with_replacement 當 with_replacement 設置為 True 時，該行為類似於 python 的 itertools.combinations。

例子：

>>> a = [1, 2, 3]
>>> list(itertools.combinations(a, r=2))
[(1, 2), (1, 3), (2, 3)]
>>> list(itertools.combinations(a, r=3))
[(1, 2, 3)]
>>> list(itertools.combinations_with_replacement(a, r=2))
[(1, 1), (1, 2), (1, 3), (2, 2), (2, 3), (3, 3)]
>>> tensor_a = torch.tensor(a)
>>> torch.combinations(tensor_a)
tensor([[1, 2],
        [1, 3],
        [2, 3]])
>>> torch.combinations(tensor_a, r=3)
tensor([[1, 2, 3]])
>>> torch.combinations(tensor_a, with_replacement=True)
tensor([[1, 1],
        [1, 2],
        [1, 3],
        [2, 2],
        [2, 3],
        [3, 3]])