(dist_inputs, distribution, coeff)
return jnp.mean(entropy_loss)
Definition of the Entropy Layer.
Definition of the Entropy Layer.
Definition of the Entropy Layer .
def EntropyLoss(dist_inputs, distribution, coeff):
  """Definition of the Entropy Layer."""
  entropy_loss = distribution.entropy(dist_inputs) * coeff
  return jnp.mean(entropy_loss)
def EntropyLoss ( dist_inputs , distribution , coeff ) : entropy_loss = distribution . entropy ( dist_inputs ) * coeff return jnp . mean ( entropy_loss )