Skip to contents

Numerical ranges on MLX devices

Source: R/create.R
mlx_arange.Rd

mlx_arange() creates evenly spaced values starting at start, stepping by step, up to and including stop (if exactly reachable). This matches R's base::seq() behavior.

Usage

mlx_arange(
  start,
  stop,
  step = 1,
  dtype = c("float32", "float64", "int8", "int16", "int32", "int64", "uint8", "uint16",
    "uint32", "uint64"),
  device = mlx_default_device()
)

Arguments

start

Starting value.

stop

Upper bound (included if exactly reachable by the step sequence).

step

Step size (defaults to 1).

dtype

Data type string. Supported types include:

  • Floating point: "float32", "float64"

  • Integer: "int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64"

  • Other: "bool", "complex64"

Not all functions support all types. See individual function documentation.

device

Execution target: supply "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Defaults to the current mlx_default_device() unless noted otherwise (helpers that act on an existing array typically reuse that array's device or stream).

Value

A 1D mlx array.

Difference from Python/C++

Unlike Python's range() and numpy.arange() which use an exclusive upper bound, mlx_arange() matches R's base::seq() by including stop only if it's exactly reachable by the step sequence. This is consistent with mlx_linspace() and mlx_slice_update(), which also follow R conventions.

See also

mlx.core.arange

Examples

mlx_arange(0, 4)        # 0, 1, 2, 3, 4
#> mlx array [5]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 0 1 2 3 4
mlx_arange(1, 5)        # 1, 2, 3, 4, 5
#> mlx array [5]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 1 2 3 4 5
mlx_arange(1, 9, 2)     # 1, 3, 5, 7, 9
#> mlx array [5]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 1 3 5 7 9
mlx_arange(1, 6, 2)     # 1, 3, 5 (6 not reachable)
#> mlx array [3]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 1 3 5