def mvn(*args, **kwargs):
  """Convenience function to efficiently construct a MultivariateNormalDiag."""
  # Faster than using `tfd.MultivariateNormalDiag`.
  return tfd.Independent(tfd.Normal(*args, **kwargs),
                         reinterpreted_batch_ndims=1)
def mvn(*args, **kwargs):
  
  
  return tfd.Independent(tfd.Normal(*args, **kwargs),
                         reinterpreted_batch_ndims=1)
def mvn ( * args , * * kwargs ) : # Faster than using `tfd.MultivariateNormalDiag`. return tfd . Independent ( tfd . Normal ( * args , * * kwargs ) , reinterpreted_batch_ndims = 1 )
Convenience function to efficiently construct a MultivariateNormalDiag.
Convenience function to efficiently construct a MultivariateNormalDiag .