当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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

计算给定张量的长度 的组合。当 with_replacement 设置为 Falseitertools.combinations_with_replacementwith_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]])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.combinations。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。