code: def negloglik(x, y, model_fn, axis=-1):
      """Negative log-likelihood."""
      return -tf.reduce_mean(model_fn(x).log_prob(y), axis=axis)
  
file_path: probability/tensorflow_probability/python/experimental/nn/losses/losses.py
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.experimental.nn import layers
  from tensorflow_probability.python.experimental.nn import variational_base
  from tensorflow_probability.python.internal.reparameterization import FULLY_REPARAMETERIZED
  from tensorflow_probability.python.monte_carlo.expectation import expectation
label_desc: Use image node_id 3 for calling a global function with example usage: negloglik(x, y, model_fn, axis) and returns: unknown