Skip to contents

Solve a system of linear equations

Source: R/linalg.R
solve.mlx.Rd

Solve a system of linear equations

Usage

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

Arguments

a

An mlx matrix of coefficients.

b

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

...

Additional arguments forwarded to the corresponding base R implementation for signature compatibility.

device

Execution target for APIs that expose a one-off device or stream override. Supply "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Ordinary array operations use the current mlx_device() instead.

Value

An mlx object containing the solution.

Details

As of MLX 0.31.1, this operation only runs on CPU. Run it inside with_device() or local_device(), or pass device = "cpu".

See also

mlx.linalg.solve

Examples

with_device("cpu", {
  a <- mlx_matrix(c(3, 1, 1, 2), 2, 2)
  b <- as_mlx(c(9, 8))
  solve(a, b)
})
#> mlx array [2]
#>   dtype: float32
#>   values:
#> [1] 2 3