Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.random.split.html below:
jax.random.split — JAX documentation
jax.random.split#
-
jax.random.split(key, num=2)[source]#
-
Splits a PRNG key into num new keys by adding a leading axis.
-
Parameters:
-
-
key (ArrayLike) – a PRNG key (from key, split, fold_in).
-
num (int | tuple[int, ...]) – optional, a positive integer (or tuple of integers) indicating the number (or shape) of keys to produce. Defaults to 2.
-
Returns:
-
An array-like object of num new PRNG keys.
-
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