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])