Función PyTorch torch.cartesian_prod
Manual de referencia de PyTorch torch
torch.cartesian_prodes una función en PyTorch utilizada para calcular el producto cartesiano. Devuelve el producto cartesiano de los tensores de entrada, es decir, el conjunto de todas las combinaciones posibles.
Definición de la función
torch.cartesian_prod(*tensors)
Ejemplo de uso
Ejemplo
import torch
# Calcular el producto cartesiano de dos tensores
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5])
result = torch.cartesian_prod(a, b)
print("a:", a)
print("b:", b)
print("Producto cartesiano:")
print(result)
# tensor([[1, 4],
# [1, 5],
# [2, 4],
# [2, 5],
# [3, 4],
# [3, 5]])
# Producto cartesiano de múltiples tensores
x = torch.tensor([1, 2])
y = torch.tensor([10, 20, 30])
z = torch.tensor([100, 200])
result = torch.cartesian_prod(x, y, z)
print("Forma del producto cartesiano de tres tensores:", result.shape)
print(result)
# Calcular el producto cartesiano de dos tensores
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5])
result = torch.cartesian_prod(a, b)
print("a:", a)
print("b:", b)
print("Producto cartesiano:")
print(result)
# tensor([[1, 4],
# [1, 5],
# [2, 4],
# [2, 5],
# [3, 4],
# [3, 5]])
# Producto cartesiano de múltiples tensores
x = torch.tensor([1, 2])
y = torch.tensor([10, 20, 30])
z = torch.tensor([100, 200])
result = torch.cartesian_prod(x, y, z)
print("Forma del producto cartesiano de tres tensores:", result.shape)
print(result)
Otras extensiones