torch.vdot

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

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

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

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

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

Примечание

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

См. также

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

Параметры:

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

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

Примечание

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

Пример:

>>> 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])