unilab.algos.mlx.common.distributions.diag_gaussian_log_prob
-
unilab.algos.mlx.common.distributions.diag_gaussian_log_prob(actions, mean, log_std)[source]
Log-probability under a diagonal Gaussian.
- Parameters:
actions (array)
mean (array)
log_std (array)
- Return type:
array