Solve triangular systems with mlx arrays

Solve triangular systems with mlx arrays

Usage

mlx_solve_triangular(a, b, upper = FALSE)

backsolve(r, x, k = NULL, upper.tri = TRUE, transpose = FALSE, ...)

# Default S3 method
backsolve(r, x, k = NULL, upper.tri = TRUE, transpose = FALSE, ...)

# S3 method for class 'mlx'
backsolve(r, x, k = NULL, upper.tri = TRUE, transpose = FALSE, ...)

Arguments

a

An mlx triangular matrix.

b

Right-hand side matrix or vector.

upper

Logical; if TRUE, a is upper triangular, otherwise lower.

r

Triangular system matrix passed to backsolve().

x

Right-hand side supplied to backsolve().

k

Number of columns of r to use.

upper.tri

Logical; indicates if r is upper triangular.

transpose

Logical; if TRUE, solve t(r) %*% x = b.

...

Additional arguments forwarded to base::backsolve().

Value

An mlx array solution.

See also

mlx.linalg.solve_triangular

Examples

a <- mlx_matrix(c(2, 1, 0, 3), 2, 2)
b <- mlx_matrix(c(1, 5), 2, 1)
mlx_solve_triangular(a, b, upper = FALSE)
#> mlx array [2 x 1]
#>   dtype: float32
#>   device: gpu
#>   values:
#>      [,1]
#> [1,]  0.5
#> [2,]  1.5