Функция PyTorch torch.addmv
Справочное руководство PyTorch torch
torch.addmvЭто функция в PyTorch, используемая для добавления результата умножения матрицы на вектор к входному вектору. Она выполняет умножение матрицы на вектор, а затем добавляет результат к входу.
Определение функции
torch.addmv(input, mat, vec, *, beta=1.0, alpha=1.0, out=None)
Parameters:
input(Tensor): Входной вектор или матрица, которые добавляются к результату.mat(Tensor): Входная матрица формы (n, m).vec(Tensor): Входной вектор формы (m,) или (m, 1).beta(float, необязательный): Коэффициент, умножаемый на вход, по умолчанию равен 1.0.alpha(float, необязательный): Коэффициент, на который умножается результат mat @ vec, по умолчанию равен 1.0.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает сумму результата умножения матрицы на вектор и входного вектора.
Пример использования
Пример
import torch
# Создать входной вектор, матрицу и вектор
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# Выполнить addmv
result = torch.addmv(input, mat, vec)
print("Форма входного вектора:", input.shape)
print("Форма матрицы:", mat.shape)
print("Форма вектора:", vec.shape)
print("Форма результата:", result.shape)
print(result)
# Создать входной вектор, матрицу и вектор
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# Выполнить addmv
result = torch.addmv(input, mat, vec)
print("Форма входного вектора:", input.shape)
print("Форма матрицы:", mat.shape)
print("Форма вектора:", vec.shape)
print("Форма результата:", result.shape)
print(result)
Результат вывода:
输入向量形状: torch.Size([3]) 矩阵形状: torch.Size([3, 4]) 向量形状: torch.Size([4]) 结果形状: torch.Size([3]) tensor([-1.0182, -0.4901, -0.9490])
Другие расширения