mv

paddle.mv ( x, vec, name\=None ) [源代码]

该op计算矩阵 x 和向量 vec 的乘积。

参数

  • x (Tensor) : 输入变量，类型为 Tensor，形状为

    mv - 图1

    ，数据类型为float32， float64。

  • vec (Tensor) : 输入变量，类型为 Tensor，形状为 [N][N]，数据类型为float32， float64。

  • name （str，可选）- 具体用法请参见 Name ，一般无需设置，默认值为None。

返回

  • Tensor，矩阵 x 和向量 vec 的乘积。

代码示例

  1. # x: [M, N], vec: [N]
  2. # paddle.mv(x, vec)  # out: [M]
  4. import numpy as np
  5. import paddle
  7. x_data = np.array([[2, 1, 3], [3, 0, 1]]).astype("float64")
  8. x = paddle.to_tensor(x_data)
  9. vec_data = np.array([3, 5, 1])
  10. vec = paddle.to_tensor(vec_data).astype("float64")
  11. out = paddle.mv(x, vec)