本文简要介绍python语言中 torch.cartesian_prod
的用法。
用法:
torch.cartesian_prod(*tensors)
*tensors-任意数量的一维张量。
一个张量相当于把所有输入的张量转成列表,对这些列表做
itertools.product
,最后把得到的列表转成张量。对给定的张量序列进行笛卡尔积。该行为类似于 python 的
itertools.product
。例子:
>>> a = [1, 2, 3] >>> b = [4, 5] >>> list(itertools.product(a, b)) [(1, 4), (1, 5), (2, 4), (2, 5), (3, 4), (3, 5)] >>> tensor_a = torch.tensor(a) >>> tensor_b = torch.tensor(b) >>> torch.cartesian_prod(tensor_a, tensor_b) tensor([[1, 4], [1, 5], [2, 4], [2, 5], [3, 4], [3, 5]])
参数:
返回:
返回类型:
相关用法
- Python PyTorch calculate_gain用法及代码示例
- Python PyTorch cat用法及代码示例
- Python PyTorch cached用法及代码示例
- Python PyTorch can_cast用法及代码示例
- Python PyTorch cholesky用法及代码示例
- Python PyTorch column_stack用法及代码示例
- Python PyTorch cumprod用法及代码示例
- Python PyTorch cov用法及代码示例
- Python PyTorch cos用法及代码示例
- Python PyTorch compute_deltas用法及代码示例
- Python PyTorch conv_transpose3d用法及代码示例
- Python PyTorch combinations用法及代码示例
- Python PyTorch conv2d用法及代码示例
- Python PyTorch cummax用法及代码示例
- Python PyTorch custom_from_mask用法及代码示例
- Python PyTorch collect_all用法及代码示例
- Python PyTorch chunk用法及代码示例
- Python PyTorch convert用法及代码示例
- Python PyTorch conv1d用法及代码示例
- Python PyTorch chain_matmul用法及代码示例
- Python PyTorch constant_用法及代码示例
- Python PyTorch context用法及代码示例
- Python PyTorch count_nonzero用法及代码示例
- Python PyTorch cdist用法及代码示例
- Python PyTorch ceil用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.cartesian_prod。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。