mindspore.ops.mv
- mindspore.ops.mv(mat, vec)[源代码]
实现矩阵 mat 和向量 vec 相乘。
如果 mat 的shape为
, vec 的的shape为 ,则输出为 的一维tensor。- 参数:
mat (Tensor) - 输入矩阵。
vec (Tensor) - 输入向量。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> output = mindspore.ops.mv(mindspore.tensor([[3., 4.], [1., 6.], [1., 3.]]), mindspore.tensor([1., 2.])) >>> print(output) [11. 13. 7.]