Функция torch.cross в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.crossЭто функция в PyTorch, используемая для вычисления векторного произведения двух трёхмерных векторов (или пакетов трёхмерных векторов). Векторное произведение создаёт новый вектор, перпендикулярный обоим входным векторам.

Определение функции

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

Пример использования

Пример

import torch

# Векторное произведение двух 3-мерных векторов
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)
# Вывод: tensor([0, 0, 1])

# Вычислить векторное произведение в пакете
a = torch.tensor([[1, 0, 0], [0, 1, 0]])
b = torch.tensor([[0, 1, 0], [1, 0, 0]])
result = torch.cross(a, b)
print("Векторное произведение для пакета:")
print(result)
# tensor([[0, 0, 1],
#         [0, 0, -1]])

# Указать размерность
a = torch.randn(3, 4, 3)
b = torch.randn(3, 4, 3)
result = torch.cross(a, b, dim=2)
print("Форма векторного произведения с указанной размерностью:", result.shape)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения