Función PyTorch torch.addmv

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

torch.addmvEs una función en PyTorch que se utiliza para agregar el resultado de la multiplicación matriz-vector al vector de entrada. Realiza la multiplicación de la matriz por el vector y luego suma el resultado con input.

Definición de la función

torch.addmv(input, mat, vec, *, beta=1.0, alpha=1.0, out=None)

Parámetros:

  • input(Tensor): vector o matriz de entrada, que se agrega al resultado.
  • mat(Tensor): matriz de entrada, de forma (n, m).
  • vec(Tensor): vector de entrada, de forma (m,) o (m, 1).
  • beta(float, opcional): coeficiente que multiplica a input, por defecto 1.0.
  • alpha(float, opcional): coeficiente que multiplica el resultado de mat @ vec, por defecto 1.0.
  • out(Tensor, opcional): tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve la suma del resultado de la multiplicación matriz-vector y el vector de entrada.

Ejemplo de uso

Ejemplo

import torch

# Crear el vector de entrada, la matriz y el vector
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)

# Ejecutar addmv
result = torch.addmv(input, mat, vec)

print("Forma del vector de entrada:", input.shape)
print("Forma de la matriz:", mat.shape)
print("Forma del vector:", vec.shape)
print("Forma del resultado:", result.shape)
print(result)

El resultado es:

输入向量形状: torch.Size([3])
矩阵形状: torch.Size([3, 4])
向量形状: torch.Size([4])
结果形状: torch.Size([3])
tensor([-1.0182, -0.4901, -0.9490])

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

Otras extensiones