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

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

jax.random.PRNGKey — JAX documentation

jax.random.PRNGKey(seed, *, impl=None)[source]#

Create a legacy PRNG key given an integer seed.

This function produces old-style legacy PRNG keys, which are arrays of dtype uint32. For more, see the note in the PRNG keys section. When possible, jax.random.key() is recommended for use instead.

The resulting key does not carry a PRNG implementation. The returned key matches the implementation given by the optional impl argument or, otherwise, determined by the jax_default_prng_impl config flag. Callers must ensure that same implementation is set as the default when passing this key as an argument to other functions (such as jax.random.split and jax.random.normal).

Parameters:

• seed (int | ArrayLike) – a 64- or 32-bit integer used as the value of the key.

• impl (PRNGSpecDesc | None | None) – optional string specifying the PRNG implementation (e.g. 'threefry2x32')

Returns:

A PRNG key, consumable by random functions as well as split and fold_in.

Return type:

Array

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