torch.mv

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

对矩阵 input 和向量 vec 执行矩阵-向量积运算。

如果 input 是一个 $(n \times m)$ 张量，vec 是大小为 $m$ 的 1-D 张量，则 out 将是大小为 $n$ 的 1-D 张量。

注意

此函数不支持广播。

参数

• input (Tensor) – 要相乘的矩阵

• vec (Tensor) – 要相乘的向量

关键字参数

out (Tensor, optional) – 输出张量。

示例

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