jax.random.uniform#

jax.random.uniform(key, shape=(), dtype=None, minval=0.0, maxval=1.0, *, out_sharding=None)[源码]#

以给定形状/数据类型采样 [minval, maxval) 范围内的均匀随机值。

参数:

  • key (ArrayLike) – 用作随机密钥的 PRNG 密钥。

  • shape (形状) – 可选，表示结果形状的非负整数元组。默认为 ()。

  • dtype (DTypeLikeFloat | None) – 可选，返回值的浮点数 dtype（如果 jax_enable_x64 为 true，则默认为 float64，否则为 float32）。

  • minval (RealArray) – 可选，范围内的最小值（包含），与 shape 兼容（默认为 0）。

  • maxval (RealArray) – 可选，范围内的最大值（不包含），与 shape 兼容（默认为 1）。

返回:

具有指定形状和 dtype 的随机数组。

返回类型:

Array