mlx.core.random.gumbel

Contents

mlx.core.random.gumbel#

gumbel(shape: Sequence[int] = [], dtype: Dtype | None = float32, key: None | Stream | Device = None, stream: array | None = None) → array#

Sample from the standard Gumbel distribution.

The values are sampled from a standard Gumbel distribution which CDF exp(-exp(-x)).

Parameters:

  • shape (list(int)) – The shape of the output.

  • dtype (Dtype, optional) – The data type of the output. Default: float32.

  • key (array, optional) – A PRNG key. Default: None.

Returns:

The array with shape shape and distributed according to the Gumbel distribution.

Return type:

array

Contents