megengine.functional.matinv¶

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

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

实际案例

>>> import numpy as np
>>> data = Tensor([[1.0, 0.0], [1.0, 1.0]])
>>> out = F.matinv(data)
>>> out.numpy()
array([[ 1.,  0.],
       [-1.,  1.]], dtype=float32)

megengine.functional.dot

megengine.functional.matmul