torch.dot

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

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

Примечание

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

Параметры:

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

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

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

Пример:

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

>>> t1, t2 = torch.tensor([0, 1]), torch.tensor([2, 3])
>>> torch.dot(t1, t2)
tensor(3)