jax.numpy.linalg.inv

jax.numpy.linalg.inv(a)[source]

Compute the (multiplicative) inverse of a matrix.

LAX-backend implementation of 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