torch.linalg.vecdot

torch.linalg.vecdot(x, y, *, dim=-1, out=None) → Tensor

Вычисляет скалярное произведение двух наборов векторов вдоль измерения.

В символах, эта функция вычисляет

$$\sum_{i=1}^n \overline{x_i}y_i.$$

по измерению dim , где $\overline{x_i}$ обозначает сопряжение для комплексных векторов, а для вещественных векторов — это тождество.

Поддерживает входные данные типов half, bfloat16, float, double, cfloat, cdouble и целых чисел. Также поддерживает векторизацию.

Параметры

• x (Тензор) – первый набор векторов формы (*, n).
• y (Тензор) – второй набор векторов формы (*, n).

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

• dim (int) – Измерение, вдоль которого вычисляется скалярное произведение. По умолчанию: -1.
• out (Тензор, необязательно) – выходной тензор. Игнорируется, если None. По умолчанию: None.

Примеры:

>>> v1 = torch.randn(3, 2)
>>> v2 = torch.randn(3, 2)
>>> linalg.vecdot(v1, v2)
tensor([ 0.3223,  0.2815, -0.1944])
>>> torch.vdot(v1[0], v2[0])
tensor(0.3223)