objax.random package¶
|
Random number generator module. |
|
Returns a |
|
Returns a |
|
Returns a |
|
Returns a |
-
class
objax.random.
Generator
(seed=0)[source]¶ Random number generator module.
-
__init__
(seed=0)[source]¶ Create a random key generator, seed is the random generator initial seed.
- Parameters
seed (int) –
-
property
key
¶ The random generator state (a tensor of 2 int32).
-
-
objax.random.
normal
(shape, *, mean=0, stddev=1, generator=<objax.random.random.Generator object>)[source]¶ Returns a
JaxArray
of shapeshape
with random numbers from a normal distribution with meanmean
and standard deviationstddev
.- Parameters
shape (Tuple[int, ..]) –
mean (float) –
stddev (float) –
generator (objax.random.random.Generator) –
-
objax.random.
randint
(shape, low, high, generator=<objax.random.random.Generator object>)[source]¶ Returns a
JaxAarray
of shapeshape
with random integers in {low, …, high-1}.- Parameters
shape (Tuple[int, ..]) –
low (int) –
high (int) –
generator (objax.random.random.Generator) –
-
objax.random.
truncated_normal
(shape, *, stddev=1, lower=-2, upper=2, generator=<objax.random.random.Generator object>)[source]¶ Returns a
JaxArray
of shapeshape
with random numbers from a normal distribution with mean 0 and standard deviationstddev
truncated by (lower
,upper
).- Parameters
shape (Tuple[int, ..]) –
stddev (float) –
lower (float) –
upper (float) –
generator (objax.random.random.Generator) –