jax.nn.initializers.ones#

jax.nn.initializers.ones(key, shape, dtype=<class 'numpy.float64'>, out_sharding=None)[source]#

一个返回由 1 组成的常量数组的初始化器。

忽略 key 参数。

>>> import jax, jax.numpy as jnp
>>> jax.nn.initializers.ones(jax.random.key(42), (3, 2), jnp.float32)
Array([[1., 1.],
       [1., 1.],
       [1., 1.]], dtype=float32)
参数:

  • key (Array)

  • shape (core.Shape)

  • dtype (DTypeLikeInexact)

  • out_sharding (OutShardingType)

返回类型:

Array