Функция torch.mv в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.mv— это функция в PyTorch, используемая для выполнения умножения матрицы на вектор. Она вычисляет произведение матрицы и вектора.

Определение функции

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

Parameters:

  • input(Tensor): Входная матрица, размер (n, m).
  • vec(Tensor): Входной вектор, размер (m,) или (m, 1).
  • out(Tensor, необязательный): Выходной тензор.

Возвращаемое значение:

  • torch.Tensor: Возвращает произведение матрицы на вектор, размер (n,).

Пример использования

Пример

import torch

# Создать матрицу и вектор
mat = torch.randn(3, 4)
vec = torch.randn(4)

# Умножение матрицы на вектор
result = torch.mv(mat, vec)

print("Форма матрицы:", mat.shape)
print("Форма вектора:", vec.shape)
print("Форма результата:", result.shape)
print("Результат:", result)

Результат вывода:

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

Пример - Вычисление слоя нейронной сети

import torch

# Симуляция полносвязного слоя нейронной сети
W = torch.randn(512, 256)  # Матрица весов
x = torch.randn(256)        # Входной вектор

# Вычислить y = W @ x
y = torch.mv(W, x)

print("Форма матрицы весов:", W.shape)
print("Форма входного вектора:", x.shape)
print("Форма выходного вектора:", y.shape)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения