def mvn(*args, **kwargs):
<EOL>return tfd.Independent(tfd.Normal(*args, **kwargs),<EOL>reinterpreted_batch_ndims=<NUM_LIT:1>)<EOL>
Convenience function to efficiently construct a MultivariateNormalDiag.