torch.cartesian_prod

torch.cartesian_prod(*tensors) [source]

Выполняет декартово произведение заданной последовательности тензоров. Поведение аналогично поведению itertools.product в Python.

Параметры

*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]])