jax.numpy.linalg.inv

jax.numpy.linalg.inv(a)

Compute the (multiplicative) inverse of a matrix.

LAX-backend implementation of numpy.linalg.inv().

Original docstring below.

Given a square matrix a, return the matrix ainv satisfying dot(a, ainv) = dot(ainv, a) = eye(a.shape[0]).

Parameters:

a ((..., M, M) array_like) – Matrix to be inverted.

Returns:

ainv – (Multiplicative) inverse of the matrix a.

Return type:

(…, M, M) ndarray or matrix