megengine.functional.dot

dot(inp1, inp2)

计算两个向量 inp1 和 inp2 的点积。输入必须是1维或标量，标量会被自动广播。在 matmul 查看更加通用的乘法。

参数

• inp1 (Tensor) – 第一个向量

• inp2 (Tensor) – 第二个向量

返回类型

Tensor

返回

输出值

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

data1 = tensor(np.arange(0, 6, dtype=np.float32))
data2 = tensor(np.arange(0, 6, dtype=np.float32))
out = F.dot(data1, data2)
print(out.numpy())

输出：

55.

megengine.functional.argmax megengine.functional.matinv