PyTorch torch.cross Funktion
PyTorch torch-Referenzhandbuch
torch.crossEs ist eine Funktion in PyTorch, die verwendet wird, um das Kreuzprodukt zweier 3-dimensionaler Vektoren (oder Stapel von 3-dimensionalen Vektoren) zu berechnen. Das Kreuzprodukt erzeugt einen neuen Vektor, der senkrecht zu beiden Eingabevektoren steht.
Funktionsdefinition
torch.cross(input, other, dim=-1)
Anwendungsbeispiel
Beispiel
import torch
# Kreuzprodukt zweier 3-dimensionaler Vektoren
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)
# Ausgabe: tensor([0, 0, 1])
# Kreuzprodukt im Batch berechnen
a = torch.tensor([[1, 0, 0], [0, 1, 0]])
b = torch.tensor([[0, 1, 0], [1, 0, 0]])
result = torch.cross(a, b)
print("Batch-Kreuzprodukt:")
print(result)
# tensor([[0, 0, 1],
# [0, 0, -1]])
# Dimension angeben
a = torch.randn(3, 4, 3)
b = torch.randn(3, 4, 3)
result = torch.cross(a, b, dim=2)
print("Form des Kreuzprodukts mit angegebener Dimension:", result.shape)
# Kreuzprodukt zweier 3-dimensionaler Vektoren
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)
# Ausgabe: tensor([0, 0, 1])
# Kreuzprodukt im Batch berechnen
a = torch.tensor([[1, 0, 0], [0, 1, 0]])
b = torch.tensor([[0, 1, 0], [1, 0, 0]])
result = torch.cross(a, b)
print("Batch-Kreuzprodukt:")
print(result)
# tensor([[0, 0, 1],
# [0, 0, -1]])
# Dimension angeben
a = torch.randn(3, 4, 3)
b = torch.randn(3, 4, 3)
result = torch.cross(a, b, dim=2)
print("Form des Kreuzprodukts mit angegebener Dimension:", result.shape)
Andere Erweiterungen