code: def mvn(*args, **kwargs):
      """Convenience function to efficiently construct a MultivariateNormalDiag."""
      # Faster than using `tfd.MultivariateNormalDiag`.
      return independent.Independent(
          normal.Normal(*args, **kwargs), reinterpreted_batch_ndims=1
      )
  
file_path: probability/tensorflow_probability/python/mcmc/eight_schools_hmc.py
imports: import time
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.mcmc import hmc
  from tensorflow_probability.python.mcmc import sample
label_desc: Use image node_id 1 for calling a global function with example usage: mvn() and returns: independent