megengine.functional.dot

dot(inp1, inp2)[源代码]

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

参数
  • inp1 (Tensor) – 第一个向量

  • inp2 (Tensor) – 第二个向量

返回类型

Tensor

返回

输出值

实际案例

>>> import numpy as np
>>> data1 = Tensor(np.arange(0, 6, dtype=np.float32))
>>> data2 = Tensor(np.arange(0, 6, dtype=np.float32))
>>> out = F.dot(data1, data2)
>>> out.numpy()
array(55., dtype=float32)