ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.distributions.reparameterized_sampling

Helper function to do reparameterized sampling if the distributions supports it.

Functions

sample(...): Sample from distribution either with reparameterized sampling or regular sampling.