Функция dot
Функция dot вычисляет скалярное произведение (внутреннее произведение)
двух одномерных тензоров. Первым и вторым параметрами функция принимает
тензоры одинаковой длины. Результатом является тензор с одним элементом
или скаляр (в зависимости от контекста).
Синтаксис
torch.dot(tensor1, tensor2)
Пример
Давайте вычислим скалярное произведение двух векторов:
import torch
t1 = torch.tensor([1, 2, 3])
t2 = torch.tensor([4, 5, 6])
res = torch.dot(t1, t2)
print(res)
Результат выполнения кода:
tensor(32)
Пример
Вычислим скалярное произведение векторов с числами с плавающей точкой:
import torch
t1 = torch.tensor([1.5, 2.5, 3.5])
t2 = torch.tensor([0.5, 1.5, 2.5])
res = torch.dot(t1, t2)
print(res)
Результат выполнения кода:
tensor(12.7500)
Пример
Если передать векторы с разными типами данных, результат будет приведён к более общему типу:
import torch
t1 = torch.tensor([1, 2, 3], dtype=torch.float)
t2 = torch.tensor([4, 5, 6], dtype=torch.int)
res = torch.dot(t1, t2)
print(res)
Результат выполнения кода:
tensor(32.)
