megengine.functional.dot

dot(inp1, inp2)[源代码]

计算两个向量 inp1inp2 的点积。输入必须是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.