mlx_rs::linalg

Function tri_inv

Source
pub fn tri_inv(a: impl AsRef<Array>, upper: Option<bool>) -> Result<Array>
Expand description

Compute the inverse of a triangular square matrix.

This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the inverse is computed for each matrix in the last two dimensions of a.