xpag.agents.rljax_agents.util.distribution

Functions

calculate_kl_divergence	Calculate KL Divergence between gaussian distributions.
evaluate_gaussian_and_tanh_log_prob	Calculate log probabilities of gaussian distributions and tanh transformation given samples.
gaussian_and_tanh_log_prob	Calculate log probabilities of gaussian distributions and tanh transformation.
gaussian_log_prob	Calculate log probabilities of gaussian distributions.
reparameterize_gaussian	Sample from gaussian distributions.
reparameterize_gaussian_and_tanh	Sample from gaussian distributions and tanh transforamation.