tfp.substrates.jax.random.spherical_uniform

Generates Tensor drawn from a uniform distribution on the sphere.

shape Vector-shaped, int Tensor representing shape of output.
dimension Scalar int Tensor, representing the dimensionality of the space where the sphere is embedded.
dtype (Optional) TF dtype representing dtype of output. Default value: tf.float32.
seed (Optional) Python integer to seed the random number generator. Default value: None (i.e., no seed).
name Python str name prefixed to Ops created by this function. Default value: None (i.e., 'random_rayleigh').

spherical_uniform Tensor with specified shape and dtype consisting of positive real values drawn from a Rayleigh distribution with specified scale.