xpag.agents.rljax_agents.util.distribution.reparameterize_gaussian#

reparameterize_gaussian(mean, log_std, key, return_log_pi=True)#

Sample from gaussian distributions.

Return type:

Tuple[Array, Array]