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 PRNG seed; see tfp.random.sanitize_seed for details. Default value: None (i.e., no seed).
name Python str name prefixed to Ops created by this function. Default value: None (i.e., 'random_spherical_uniform').

spherical_uniform Tensor with specified shape and dtype consisting of real values drawn from a spherical uniform distribution.