jax.nn.initializers.uniform#

jax.nn.initializers.uniform(scale=0.01, dtype=<class 'jax.numpy.float64'>)[原始碼]#

建構一個初始化器，其傳回實數均勻分佈的隨機陣列。

參數:

  • scale (RealNumeric) – 選填；隨機分佈的上限。

  • dtype (DTypeLikeInexact) – 選填；初始化器的預設 dtype。

傳回:

一個初始化器，其傳回值在範圍 [0, scale) 內均勻分佈的陣列。

傳回類型:

Initializer

>>> import jax, jax.numpy as jnp
>>> initializer = jax.nn.initializers.uniform(10.0)
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)  
Array([[7.298188 , 8.691938 , 8.7230015],
       [2.0818567, 1.8662417, 5.5022564]], dtype=float32)