megengine.functional.matinv¶

计算输入矩阵的逆；输入矩阵的形状必须为 […, n, n] 的形式。

参数: inp (Tensor) – 输入张量。
返回类型: Tensor
返回: 输出张量。

实际案例

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

data = tensor([[1.0, 0.0], [1.0, 1.0]])
out = F.matinv(data)
print(out.numpy())

输出：

[[ 1.  0.]
 [-1.  1.]]

megengine.functional.dot

megengine.functional.matmul