paddle / mv

mv¶

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

该op计算矩阵 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 numpy as np
import paddle

x_data = np.array([[2, 1, 3], [3, 0, 1]]).astype("float64")
x = paddle.to_tensor(x_data)
vec_data = np.array([3, 5, 1])
vec = paddle.to_tensor(vec_data).astype("float64")
out = paddle.mv(x, vec)

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle / mv

mv¶

参数¶

返回¶

代码示例¶