torch.mv

torch.mv(input, vec, *, out=None) → Tensor

Вычисляет произведение матрицы input на вектор vec.

Если input — это тензор размера $(n \times m)$, vec — это одномерный тензор размера $m$, out будет одномерным тензором размера $n$.

Примечание

Эта функция не выполняет широковещание.

Параметры:

• input (Tensor) – матрица для умножения
• vec (Tensor) – вектор для умножения

Именованные аргументы:

out (Tensor, необязательный) – выходной тензор.

Пример:

>>> mat = torch.randn(2, 3)
>>> vec = torch.randn(3)
>>> torch.mv(mat, vec)
tensor([ 1.0404, -0.6361])