Generate raw random bits on MLX arrays

Generate raw random bits on MLX arrays

Usage

mlx_key_bits(dim, width = 4L, key = NULL, device = mlx_default_device())

Arguments

dim

Integer vector specifying array dimensions (shape).

width

Number of bytes per element (default 4 = 32 bits). Must be positive.

key

Optional mlx key array. If omitted, MLX's default generator is used.

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 of unsigned integers filled with random bits.

See also

mlx.core.random.bits

Examples

k <- mlx_key(12)
raw_bits <- mlx_key_bits(c(4, 4), key = k)