mlx_rs::linalg

Function solve_device

Source 
pub fn solve_device(
    a: impl AsRef<Array>,
    b: impl AsRef<Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

Compute the solution to a system of linear equations AX = B

§Params

  • a: input array
  • b: input array
  • stream: stream to execute the operation

§Returns

The unique solution to the system AX = B