torch.dot

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

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

Примечание

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

Параметры

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

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

out (Тензор, необязательно) – тензор результата.

Пример:

>>> torch.dot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)