Module: tfp.substrates.jax.random

TensorFlow Probability random samplers/utilities.

Functions

rademacher(...): Generates Tensor consisting of -1 or +1, chosen uniformly at random.

rayleigh(...): Generates Tensor of positive reals drawn from a Rayleigh distributions.

sanitize_seed(...): Map various PRNG seed flavors to a seed Tensor.

spherical_uniform(...): Generates Tensor drawn from a uniform distribution on the sphere.

split_seed(...): Splits a seed into n derived seeds.