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)