Skip to main content

gumbel

Function gumbel 

Source
pub fn gumbel<'a>(
    shape: impl IntoShape,
    dtype: DType,
    key: impl Into<Option<&'a Array>>,
) -> Array
Expand description

Gumbel samples of shape and floating-point dtype, as MLX: −log(−log(u)) for u uniform in [0, 1).