def preferred_move(self):
    """Preferred move - the mean of selected moves."""
    def f(dist_inputs, values):
      del values
      return rl_layers.PreferredMove(dist_inputs, self._policy_dist.sample)
    return tl.Fn('PreferredMove', f)
Preferred move - the mean of selected moves.
trax/rl/actor_critic_joint.py