Функция PyTorch torch.vdot
Справочное руководство Pytorch torch
torch.vdot— это функция в PyTorch, используемая для вычисления скалярного произведения двух векторов (также известного как скалярное произведение векторов). В отличие отtorch.dotотличается,torch.vdotможет обрабатывать комплексные векторы и автоматически умножает каждый элемент первого вектора на сопряжённый элемент второго вектора.
Определение функции
torch.vdot(input, other)
Пример использования
Пример
import torch
# Вещественный вектор
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print("результат vdot:", result)
# 1*4 + 2*5 + 3*6 = 32
# Комплексный вектор
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print("Результат vdot для комплексных чисел:", result_complex)
# Вещественный вектор
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.vdot(a, b)
print("результат vdot:", result)
# 1*4 + 2*5 + 3*6 = 32
# Комплексный вектор
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])
result_complex = torch.vdot(c, d)
print("Результат vdot для комплексных чисел:", result_complex)
Результат вывода:
vdot 结果: tensor(32) 复数 vdot 结果: tensor(70.-6.j)
Другие расширения