code: def EntropyLoss(dist_inputs, distribution, coeff):
      """Definition of the Entropy Layer."""
      entropy_loss = distribution.entropy(dist_inputs) * coeff
      return jnp.mean(entropy_loss)
  
file_path: trax/trax/rl/rl_layers.py
imports: from trax.fastmath import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: EntropyLoss(dist_inputs, distribution, coeff) and returns: jnp