tfp.substrates.jax.random.rademacher

Generates Tensor consisting of -1 or +1, chosen uniformly at random.

tfp.substrates.jax.random.rademacher(
    shape, dtype=tf.float32, seed=None, name=None
)

For more details, see Rademacher distribution.

shape Vector-shaped, int Tensor representing shape of output.
dtype (Optional) TF dtype representing dtype of output.
seed PRNG seed; see tfp.random.sanitize_seed for details.
name Python str name prefixed to Ops created by this function. Default value: None (i.e., 'random_rademacher').

rademacher Tensor with specified shape and dtype consisting of -1 or +1 chosen uniformly-at-random.