megengine.functional.matinv¶

Computes the inverse of a batch of matrices; input must has shape […, n, n].

Parameters: inp (Tensor) – input tensor.
Return type: Tensor
Returns: output tensor.

Examples

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