megengine.functional.matinv

matinv(inp)[源代码]

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

参数

inp – 输入张量。

返回

输出张量。

例如:

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.]]