(self)
return tl.Fn('AdvantageNorm', f)
Norm of advantages.
Norm of advantages.
Norm of advantages .
def advantage_norm(self):
    """Norm of advantages."""
    def f(dist_inputs, values, returns):
      del dist_inputs
      return jnp.linalg.norm(returns - values)
    return tl.Fn('AdvantageNorm', f)
def advantage_norm ( self ) : def f ( dist_inputs , values , returns ) : del dist_inputs return jnp . linalg . norm ( returns - values ) return tl . Fn ( 'AdvantageNorm' , f )