Fonction torch.cross de PyTorch

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

torch.crossIl s'agit d'une fonction de PyTorch utilisée pour calculer le produit vectoriel de deux vecteurs 3D (ou de lots de vecteurs 3D). Le produit vectoriel génère un nouveau vecteur perpendiculaire aux deux vecteurs d'entrée.

Définition de fonction

torch.cross(input, other, dim=-1)

Exemple d'utilisation

Exemple

import torch

# Produit vectoriel de deux vecteurs tridimensionnels
a = torch.tensor([1, 0, 0])
b = torch.tensor([0, 1, 0])
c = torch.cross(a, b)
print("a:", a)
print("b:", b)
print("a x b:", c)
# Sortie : tensor([0, 0, 1])

# Calculer le produit vectoriel par lots
a = torch.tensor([[1, 0, 0], [0, 1, 0]])
b = torch.tensor([[0, 1, 0], [1, 0, 0]])
result = torch.cross(a, b)
print("Produit vectoriel par lots :")
print(result)
# tensor([[0, 0, 1],
#         [0, 0, -1]])

# Spécifier la dimension
a = torch.randn(3, 4, 3)
b = torch.randn(3, 4, 3)
result = torch.cross(a, b, dim=2)
print("Forme du produit vectoriel avec la dimension spécifiée :", result.shape)

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

Autres extensions