jax.random.uniform#

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

Sample uniform random values in [minval, maxval) with given shape/dtype.

Parameters:
Return type:

Array

Returns:

A random array with the specified shape and dtype.