Функция PyTorch torch.dot

Pytorch torch 参考手册Справочное руководство PyTorch torch

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

Определение функции

torch.dot(input, other, out)

Пример использования

Пример

import torch

a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

# Скалярное произведение: 1*4 + 2*5 + 3*6 = 32
result = torch.dot(a, b)

print(result)

Вывод:

tensor(32)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения