jax-js
    Preparing search index...
    • Solve a linear system of equations.

      This solves a (batched) linear system of equations a @ x = b for x given a and b. If a is singular, this will return nan or inf values.

      Parameters

      • a: ArrayLike

        Coefficient matrix of shape (..., N, N).

      • b: ArrayLike

        Values of shape (N,) or (..., N, M).

      Returns Array

      Solution x of shape (..., N) or (..., N, M).