jax.numpy.linalg.inv#
- jax.numpy.linalg.inv(a)[source]#
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