jax.random.uniform

目录

jax.random.uniform#

jax.random.uniform(key, shape=(), dtype=<class 'float'>, minval=0.0, maxval=1.0)[源代码][源代码]#

在给定的形状/数据类型下，生成 [minval, maxval) 范围内的均匀随机值。

参数:

  • key (KeyArrayLike) – 一个用作随机密钥的 PRNG 密钥。

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

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

  • minval (RealArray) – 可选，一个与范围形状广播兼容的最小（包含）值（默认值为0）。

  • maxval (RealArray) – 可选，一个与范围形状广播兼容的最大（排除）值（默认值为1）。

返回:

具有指定形状和数据类型的随机数组。

返回类型:

Array

目录