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


Python PyTorch cartesian_prod用法及代码示例


本文简要介绍python语言中 torch.cartesian_prod 的用法。

用法:

torch.cartesian_prod(*tensors)

参数

*tensors-任意数量的一维张量。

返回

一个张量相当于把所有输入的张量转成列表,对这些列表做itertools.product,最后把得到的列表转成张量。

返回类型

Tensor

对给定的张量序列进行笛卡尔积。该行为类似于 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]])

相关用法


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