Solve a system of linear equations

Solve a system of linear equations

Usage

# S3 method for class 'mlx'
solve(a, b = NULL, ...)

Arguments

a

An mlx matrix (the coefficient matrix)

b

An mlx vector or matrix (the right-hand side). If omitted, computes the matrix inverse.

...

Additional arguments (for compatibility with base::solve)

Value

An mlx object containing the solution.

See also

mlx.linalg.solve

Examples

a <- mlx_matrix(c(3, 1, 1, 2), 2, 2)
b <- as_mlx(c(9, 8))
solve(a, b)
#> mlx array [2]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 2 3