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)

megengine.functional.normalize

megengine.functional.matinv