mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
71 lines
1.0 KiB
ReStructuredText
71 lines
1.0 KiB
ReStructuredText
``jax.random`` module
|
|
=====================
|
|
|
|
.. automodule:: jax.random
|
|
|
|
API Reference
|
|
-------------
|
|
|
|
Key Creation & Manipulation
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
PRNGKey
|
|
key
|
|
key_data
|
|
wrap_key_data
|
|
fold_in
|
|
split
|
|
clone
|
|
|
|
Random Samplers
|
|
~~~~~~~~~~~~~~~
|
|
|
|
.. Generate the list of callable members:
|
|
>>> from jax import random
|
|
>>> fns = (x for x in sorted(dir(random)) if callable(getattr(random, x)))
|
|
>>> print('\n'.join(' ' + x for x in fns)) # doctest: +SKIP
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
ball
|
|
bernoulli
|
|
beta
|
|
binomial
|
|
bits
|
|
categorical
|
|
cauchy
|
|
chisquare
|
|
choice
|
|
dirichlet
|
|
double_sided_maxwell
|
|
exponential
|
|
f
|
|
gamma
|
|
generalized_normal
|
|
geometric
|
|
gumbel
|
|
laplace
|
|
loggamma
|
|
logistic
|
|
lognormal
|
|
maxwell
|
|
multivariate_normal
|
|
normal
|
|
orthogonal
|
|
pareto
|
|
permutation
|
|
poisson
|
|
rademacher
|
|
randint
|
|
rayleigh
|
|
t
|
|
triangular
|
|
truncated_normal
|
|
uniform
|
|
wald
|
|
weibull_min
|