code: def PreferredMove(dist_inputs, sample):
      """Definition of the preferred move."""
      preferred_moves = sample(dist_inputs, temperature=0.0)
      return jnp.mean(preferred_moves)
  
file_path: trax/trax/rl/rl_layers.py
imports: from trax.fastmath import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: PreferredMove(dist_inputs, sample) and returns: jnp