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.

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