Функция torch.cartesian_prod в PyTorch
Справочное руководство PyTorch torch
torch.cartesian_prodЭто функция в PyTorch, используемая для вычисления декартова произведения. Она возвращает декартово произведение входных тензоров, т. е. множество всех возможных комбинаций.
Определение функции
torch.cartesian_prod(*tensors)
Пример использования
Пример
import torch
# Вычислить декартово произведение двух тензоров
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5])
result = torch.cartesian_prod(a, b)
print("a:", a)
print("b:", b)
print("Декартово произведение:")
print(result)
# tensor([[1, 4],
# [1, 5],
# [2, 4],
# [2, 5],
# [3, 4],
# [3, 5]])
# Декартово произведение нескольких тензоров
x = torch.tensor([1, 2])
y = torch.tensor([10, 20, 30])
z = torch.tensor([100, 200])
result = torch.cartesian_prod(x, y, z)
print("Форма декартова произведения трёх тензоров:", result.shape)
print(result)
# Вычислить декартово произведение двух тензоров
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5])
result = torch.cartesian_prod(a, b)
print("a:", a)
print("b:", b)
print("Декартово произведение:")
print(result)
# tensor([[1, 4],
# [1, 5],
# [2, 4],
# [2, 5],
# [3, 4],
# [3, 5]])
# Декартово произведение нескольких тензоров
x = torch.tensor([1, 2])
y = torch.tensor([10, 20, 30])
z = torch.tensor([100, 200])
result = torch.cartesian_prod(x, y, z)
print("Форма декартова произведения трёх тензоров:", result.shape)
print(result)
Другие расширения