torch.cross

torch.cross(input, other, dim=None, *, out=None) → Tensor

Возвращает векторное произведение векторов в измерении dim для input и other.

Поддерживает входные данные типов float, double, cfloat и cdouble. Также поддерживает пакеты векторов, для которых вычисляется произведение вдоль измерения dim. В этом случае выходные данные имеют те же размерности пакета, что и входные.

Если dim не указано, оно по умолчанию равно первому размеру с размером 3. Обратите внимание, что это может быть неожиданно.

См. также

torch.linalg.cross(), которая требует указания dim (по умолчанию -1).

Предупреждение

Эта функция может измениться в будущих выпусках PyTorch, чтобы соответствовать поведению по умолчанию в torch.linalg.cross(). Мы рекомендуем использовать torch.linalg.cross().

Параметры:

• input (Тензор) – входной тензор.
• other (Тензор) – второй входной тензор
• dim (int, необязательно) – измерение, в котором нужно выполнить векторное произведение.

Ключевые аргументы:

out (Тензор, необязательно) – выходной тензор.

Пример:

>>> a = torch.randn(4, 3)
>>> a
tensor([[-0.3956,  1.1455,  1.6895],
        [-0.5849,  1.3672,  0.3599],
        [-1.1626,  0.7180, -0.0521],
        [-0.1339,  0.9902, -2.0225]])
>>> b = torch.randn(4, 3)
>>> b
tensor([[-0.0257, -1.4725, -1.2251],
        [-1.1479, -0.7005, -1.9757],
        [-1.3904,  0.3726, -1.1836],
        [-0.9688, -0.7153,  0.2159]])
>>> torch.cross(a, b, dim=1)
tensor([[ 1.0844, -0.5281,  0.6120],
        [-2.4490, -1.5687,  1.9792],
        [-0.8304, -1.3037,  0.5650],
        [-1.2329,  1.9883,  1.0551]])
>>> torch.cross(a, b)
tensor([[ 1.0844, -0.5281,  0.6120],
        [-2.4490, -1.5687,  1.9792],
        [-0.8304, -1.3037,  0.5650],
        [-1.2329,  1.9883,  1.0551]])