megengine.functional.dot

dot(inp1, inp2)

计算两个向量 inp1 和 inp2 的点积。输入必须是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)