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 и целочисленных типов. Также поддерживается broadcasting.

Параметры:

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

Именованные аргументы:

• dim (int) – измерение, вдоль которого вычисляется скалярное произведение. По умолчанию: -1.
• out (Tensor, optional) – выходной тензор. Игнорируется, если 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)