本文简要介绍python语言中 torch.combinations
的用法。
用法:
torch.combinations(input, r=2, with_replacement=False) → seq
一个张量相当于把所有输入的张量转成列表,对这些列表做
itertools.combinations
或者itertools.combinations_with_replacement
,最后把得到的列表转成张量。计算给定张量的长度
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]])
参数:
返回:
返回类型:
相关用法
- Python PyTorch compute_deltas用法及代码示例
- Python PyTorch complex用法及代码示例
- Python PyTorch column_stack用法及代码示例
- Python PyTorch cov用法及代码示例
- Python PyTorch cos用法及代码示例
- Python PyTorch conv_transpose3d用法及代码示例
- Python PyTorch conv2d用法及代码示例
- Python PyTorch collect_all用法及代码示例
- Python PyTorch convert用法及代码示例
- Python PyTorch conv1d用法及代码示例
- Python PyTorch constant_用法及代码示例
- Python PyTorch context用法及代码示例
- Python PyTorch count_nonzero用法及代码示例
- Python PyTorch cosh用法及代码示例
- Python PyTorch cosine_similarity用法及代码示例
- Python PyTorch conv3d用法及代码示例
- Python PyTorch cond用法及代码示例
- Python PyTorch conj_physical用法及代码示例
- Python PyTorch conj用法及代码示例
- Python PyTorch corrcoef用法及代码示例
- Python PyTorch copysign用法及代码示例
- Python PyTorch conv_transpose2d用法及代码示例
- Python PyTorch conv_transpose1d用法及代码示例
- Python PyTorch cholesky用法及代码示例
- Python PyTorch cumprod用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.combinations。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。