torch.vdot

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

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

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

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

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

Примечание

В отличие от NumPy’s vdot, torch.vdot преднамеренно поддерживает только вычисление скалярного произведения двух одномерных тензоров с одинаковым количеством элементов.

См. также

torch.linalg.vecdot() вычисляет скалярное произведение двух наборов векторов вдоль измерения.

Параметры

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

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

Примечание

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

Пример:

>>> torch.vdot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)
>>> a = torch.tensor((1 +2j, 3 - 1j))
>>> b = torch.tensor((2 +1j, 4 - 0j))
>>> torch.vdot(a, b)
tensor([16.+1.j])
>>> torch.vdot(b, a)
tensor([16.-1.j])