mlx_rs::linalg

Function tri_inv_device

Source
pub fn tri_inv_device(
    a: impl AsRef<Array>,
    upper: Option<bool>,
    stream: impl AsRef<Stream>,
) -> 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.