Skip to contents

Sample from the Gumbel distribution on mlx arrays

Source: R/random.R
mlx_rand_gumbel.Rd

Sample from the Gumbel distribution on mlx arrays

Usage

mlx_rand_gumbel(
  dim,
  dtype = c("float32", "float64"),
  device = mlx_default_device()
)

Arguments

dim

Integer vector specifying array dimensions (shape).

dtype

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

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 with Gumbel-distributed entries.

See also

mlx.core.random.gumbel

Examples

samples <- mlx_rand_gumbel(c(2, 3))