Evenly spaced ranges on MLX devices

mlx_linspace() creates num evenly spaced values from start to stop, inclusive. Unlike mlx_arange(), you specify how many samples you want rather than the step size.

Usage

mlx_linspace(
  start,
  stop,
  num = 50L,
  dtype = c("float32", "float64"),
  device = mlx_default_device()
)

Arguments

start

Starting value.

stop

Final value (inclusive).

num

Number of samples to generate.

dtype

MLX dtype ("float32" or "float64").

device

Execution target: provide "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Defaults to the current mlx_default_device().

Value

A 1D mlx array.

See also

mlx.core.linspace

Examples

mlx_linspace(0, 1, num = 5)
#> mlx array [5]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 0.00 0.25 0.50 0.75 1.00