Fonction torch.cartesian_prod de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.cartesian_prodC'est une fonction dans PyTorch utilisée pour calculer le produit cartésien. Elle renvoie le produit cartésien des tenseurs d'entrée, c'est-à-dire l'ensemble de toutes les combinaisons possibles.

Définition de fonction

torch.cartesian_prod(*tensors)

Exemple d'utilisation

Exemple

import torch

# Calculer le produit cartésien de deux tenseurs
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5])
result = torch.cartesian_prod(a, b)
print("a:", a)
print("b:", b)
print("Produit cartésien :")
print(result)
# tensor([[1, 4],
#         [1, 5],
#         [2, 4],
#         [2, 5],
#         [3, 4],
#         [3, 5]])

# Produit cartésien de plusieurs tenseurs
x = torch.tensor([1, 2])
y = torch.tensor([10, 20, 30])
z = torch.tensor([100, 200])
result = torch.cartesian_prod(x, y, z)
print("Forme du produit cartésien de trois tenseurs :", result.⟦⟧shapeshape)
print(result)

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions