mindspore.ops.mv

View Source On Gitee
mindspore.ops.mv(mat, vec)[source]

Multiplies matrix mat and vector vec.

If mat is a Tensor with \((N, M)\), vec is a 1-D Tensor of size \(M\), out will be 1-D of size \(N\).

Parameters

  • mat (Tensor) – Input matrix of shape \((N, M)\).

  • vec (Tensor) – Input vector of shape \((M,)\).

Returns

Tensor, the shape of the output Tensor is \((N,)\).

Raises

  • TypeError – If mat or vec is not a Tensor.

  • ValueError – If mat is not a 2-D Tensor or vec is not a 1-D Tensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> mat = Tensor(np.array([[3., 4.], [1., 6.], [1., 3.]]).astype(np.float32))
>>> vec = Tensor(np.array([1., 2.]).astype(np.float32))
>>> output = ops.mv(mat, vec)
>>> print(output)
[11. 13. 7.]