torch.cartesian_prod¶
- torch.cartesian_prod(*tensors)[source]¶
对给定的张量序列执行笛卡尔积。其行为类似于 Python 的 itertools.product。
- 参数
*tensors (Tensor) – 任意数量的一维张量。
- 返回值
一个张量,等效于将所有输入张量转换为列表,对这些列表执行 itertools.product,最后将结果列表转换为张量。
- 返回类型
示例
>>> 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]])