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
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 anmlx_streamcreated viamlx_new_stream(). Defaults to the currentmlx_default_device()unless noted otherwise (helpers that act on an existing array typically reuse that array's device or stream).
Examples
samples <- mlx_rand_gumbel(c(2, 3))