torch.addmv¶
- torch.addmv(input, mat, vec, *, beta=1, alpha=1, out=None) Tensor ¶
执行矩阵
mat
与向量vec
的矩阵向量乘积。向量input
将添加到最终结果中。如果
mat
是一个 张量,vec
是一个大小为 m 的一维张量,那么input
必须与大小为 n 的一维张量可广播,out
将是大小为 n 的一维张量。alpha
和beta
分别是mat
和vec
之间的矩阵向量乘积以及添加的张量input
的缩放因子。如果
beta
为 0,则将忽略input
,并且其中的 nan 和 inf 不会传播。对于类型为 FloatTensor 或 DoubleTensor 的输入,参数
beta
和alpha
必须是实数,否则它们应该是整数。- 参数
- 关键字参数
beta (Number, optional) –
input
的乘数 ()alpha (数字, 可选) – 的乘数 ()
out (张量, 可选) – 输出张量。
示例
>>> M = torch.randn(2) >>> mat = torch.randn(2, 3) >>> vec = torch.randn(3) >>> torch.addmv(M, mat, vec) tensor([-0.3768, -5.5565])