Skip to contents

Sample from a uniform distribution on mlx arrays

Source: R/random.R
mlx_rand_uniform.Rd

Sample from a uniform distribution on mlx arrays

Usage

mlx_rand_uniform(
  dim,
  min = 0,
  max = 1,
  dtype = c("float32", "float64"),
  device = mlx_default_device()
)

Arguments

dim

Integer vector specifying array dimensions (shape).

min

Lower bound for the uniform distribution.

max

Upper bound for the uniform distribution.

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

An mlx array whose entries are sampled uniformly.

See also

mlx.core.random.uniform

Examples

noise <- mlx_rand_uniform(c(2, 2), min = -1, max = 1)