Función torch.vdot de PyTorch

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.vdotEs una función en PyTorch utilizada para calcular el producto punto de dos vectores (también conocido como producto escalar vectorial).torch.dotA diferencia de lo anterior,torch.vdotpuede manejar vectores complejos y multiplica automáticamente cada elemento del primer vector por el elemento conjugado del segundo vector.

Definición de la función

torch.vdot(input, other)

Ejemplo de uso

Ejemplo

import torch

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

result = torch.vdot(a, b)

print("Resultado de vdot:", result)
# 1*4 + 2*5 + 3*6 = 32

# Vector complejo
c = torch.tensor([1+2j, 3+4j])
d = torch.tensor([5+6j, 7+8j])

result_complex = torch.vdot(c, d)

print("Resultado de vdot complejo:", result_complex)

El resultado de salida es:

vdot 结果: tensor(32)
复数 vdot 结果: tensor(70.-6.j)

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones