mindspore.ops.addmv
- mindspore.ops.addmv(input, mat, vec, *, beta=1, alpha=1)[源代码]
将矩阵 mat 和 向量 vec 相乘,结果与输入 input 相加。
说明
如果 mat 是一个大小为
的tensor,且 vec 是一个大小为 的一维tensor,那么 input 必须是可广播的,并且是一个大小为 的一维tensor。若 beta 为0,那么 input 将被忽略。
- 参数:
input (Tensor) - 输入tensor。
mat (Tensor) - 将被乘的矩阵tensor。
vec (Tensor) -将被乘的向量tensor。
- 关键字参数:
beta (scalar[int, float, bool], 可选) - input 的尺度因子,默认
1
。alpha (scalar[int, float, bool], 可选) - ( mat @ vec )的尺度因子,默认
1
。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> input = mindspore.tensor([2., 3.]) >>> mat = mindspore.tensor([[2., 5., 3.], [4., 2., 2.]]) >>> vec = mindspore.tensor([3., 2., 4.]) >>> output = mindspore.ops.addmv(input, mat, vec) >>> print(output) [30. 27.]