mlx.core.random.gumbel

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

從標準 Gumbel 分佈取樣。

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

參數:

• 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.

回傳:

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

回傳型別:

array