(dist_inputs, sample)
return jnp.mean(preferred_moves)
Definition of the preferred move.
Definition of the preferred move.
Definition of the preferred move .
def PreferredMove(dist_inputs, sample):
  """Definition of the preferred move."""
  preferred_moves = sample(dist_inputs, temperature=0.0)
  return jnp.mean(preferred_moves)
def PreferredMove ( dist_inputs , sample ) : preferred_moves = sample ( dist_inputs , temperature = 0.0 ) return jnp . mean ( preferred_moves )