mv¶

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

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

参数¶

x (Tensor) - 输入变量，类型为 Tensor，形状为 \([M, N]\)，数据类型为 float32、float64。

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

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

返回¶

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

代码示例¶

          # x: [M, N], vec: [N]
# paddle.mv(x, vec)  # out: [M]

import paddle

x = paddle.to_tensor([[2, 1, 3], [3, 0, 1]]).astype("float64")
vec = paddle.to_tensor([3, 5, 1]).astype("float64")
out = paddle.mv(x, vec)
print(out)
# Tensor(shape=[2], dtype=float64, place=Place(cpu), stop_gradient=True,
#        [14., 10.])

         