快捷方式

torch.cartesian_prod

torch.cartesian_prod(*tensors)[source]

对给定的张量序列执行笛卡尔积。其行为类似于 Python 的 itertools.product

参数

*tensors (Tensor) – 任意数量的一维张量。

返回值

一个张量,等效于将所有输入张量转换为列表,对这些列表执行 itertools.product,最后将结果列表转换为张量。

返回类型

Tensor

示例

>>> import itertools
>>> 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 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源