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