Skip to contents

Evenly spaced ranges on MLX devices

Source: R/create.R
mlx_linspace.Rd

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"))

Arguments

start

Starting value.

stop

Final value (inclusive).

num

Number of samples to generate.

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.

Value

A 1D mlx array.

Details

MLX does not support float64 operations on GPU. When this function creates a float64 array or converts one back to R, Rmlx temporarily switches only that internal creation or layout work to CPU. Later operations on the returned array still use the current mlx_device().

See also

mlx.core.linspace

Examples

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