(self)
return tl.Fn('LogProbsMean', f)
Mean of log_probs aka dist_inputs.
Mean of log_probs aka dist_inputs.
Mean of log_probs aka dist_inputs .
def log_probs_mean(self):
    """Mean of log_probs aka dist_inputs."""
    def f(dist_inputs, values):
      del values
      return jnp.mean(dist_inputs)
    return tl.Fn('LogProbsMean', f)
def log_probs_mean ( self ) : def f ( dist_inputs , values ) : del values return jnp . mean ( dist_inputs ) return tl . Fn ( 'LogProbsMean' , f )