Función PyTorch torch.mv

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

torch.mvEs la función en PyTorch utilizada para ejecutar la multiplicación de matriz por vector. Calcula el producto de una matriz y un vector.

Definición de la función

torch.mv(input, vec, out=None)

Parámetros:

  • input(Tensor): la matriz de entrada, con forma (n, m).
  • vec(Tensor): el vector de entrada, con forma (m,) o (m, 1).
  • out(Tensor, opcional): el tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el producto matriz-vector, con forma (n,).

Ejemplo de uso

Ejemplo

import torch

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

# Multiplicación de matriz por vector
result = torch.mv(mat, vec)

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

El resultado de salida es:

矩阵形状: torch.Size([3, 4])
向量形状: torch.Size([4])
结果形状: torch.Size([3])
tensor([-0.1861,  0.2482, -0.4375])

Ejemplo - Cálculo de capa de red neuronal

import torch

# Simular la capa completamente conectada de una red neuronal
W = torch.randn(512, 256)  # Matriz de pesos
x = torch.randn(256)        # Vector de entrada

# Calcular y = W @ x
y = torch.mv(W, x)

print("Forma de la matriz de pesos:", W.shape)
print("Forma del vector de entrada:", x.shape)
print("Forma del vector de salida:", y.shape)

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

Otras extensiones