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
Examples
a <- as_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