Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.random.wrap_key_data.html below:

jax.random.wrap_key_data — JAX documentation

jax.random.wrap_key_data(key_bits_array, *, impl=None)[source]#

Wrap an array of key data bits into a PRNG key array.

Parameters:

• key_bits_array (Array) – a uint32 array with trailing shape corresponding to the key shape of the PRNG implementation specified by impl.

• impl (PRNGSpecDesc | None | None) – optional, specifies a PRNG implementation, as in random.key.

Returns:

A PRNG key array, whose dtype is a subdtype of jax.dtypes.prng_key

corresponding to impl, and whose shape equals the leading shape of key_bits_array.shape up to the key bit dimensions.

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.3