评价此页

torch.mv#

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

执行矩阵 input 与向量 vec 的矩阵-向量乘积。

如果 input 是一个 (n×m)(n \times m) 张量,vec 是一个大小为 mm 的一维张量,那么 out 将是一个大小为 nn 的一维张量。

注意

此函数不执行广播

参数
  • 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])