Función PyTorch torch.dot
Manual de referencia de Pytorch torch
torch.dotEs una función en PyTorch para calcular el producto punto de dos vectores.
Definición de la función
torch.dot(input, other, out)
Ejemplo de uso
Ejemplo
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Producto punto: 1*4 + 2*5 + 3*6 = 32
result = torch.dot(a, b)
print(result)
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Producto punto: 1*4 + 2*5 + 3*6 = 32
result = torch.dot(a, b)
print(result)
El resultado de salida es:
tensor(32)
Otras extensiones