Module random

Collection of functions related to random number generation

Structs

RandomState

Random state factory

Enums

ShapeOrCount

Shape or count for the categorical distribution.

Functions

bernoulli

Generate Bernoulli random values with a given p value.

bernoulli_device

Generate Bernoulli random values with a given p value.

categorical

Sample from a categorical distribution.

categorical_device

Sample from a categorical distribution.

gumbel

Sample from the standard Gumbel distribution.

gumbel_device

Sample from the standard Gumbel distribution.

key

Get a PRNG key from a seed.

multivariate_normal

Generate jointly-normal random samples given a mean and covariance.

multivariate_normal_device

Generate jointly-normal random samples given a mean and covariance.

normal

Generate normally distributed random numbers.

normal_device

Generate normally distributed random numbers.

randint

Generate random integers from the given interval (lower: and upper:).

randint_device

Generate random integers from the given interval (lower: and upper:).

seed

Seed the random number generator.

split

Split a PRNG key into two keys and return a tuple.

split_device

Split a PRNG key into two keys and return a tuple.

truncated_normal

Generate values from a truncated normal distribution between low and high.

truncated_normal_device

Generate values from a truncated normal distribution between low and high.

uniform

Generate uniformly distributed random numbers. The values are sampled uniformly in the half-open interval [lower, upper). The lower and upper bound can be scalars or arrays and must be broadcastable to shape.

uniform_device

Generate uniformly distributed random numbers. The values are sampled uniformly in the half-open interval [lower, upper). The lower and upper bound can be scalars or arrays and must be broadcastable to shape.

with_random_state

Use the given random state for the scope of f