code: def testCorrelatedMVNManySteps(self):
      # Assert that thinning inside the autobatched nuts gives better optimal
      # utilization, in the sense that the number of leapfrogs computed is forced
      # by the length of some batch member's computation.
      kernel = self._correlated_mvn_nuts(
          dim=10, step_size=0.4, num_steps=10
      )
      _, extra_ = sample.sample_chain(
          num_results=1,
          num_burnin_steps=0,
          current_state=[tf.zeros([20, 10])],
          kernel=kernel,
          parallel_iterations=1,
      )
      extra = self.evaluate(extra_)
      for c in extra.leapfrogs_computed[-1]:
          self.assertEqual(c, extra.leapfrogs_taken.max())
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import chain
  from tensorflow_probability.python.bijectors import invert
  from tensorflow_probability.python.bijectors import scale_matvec_tril
  from tensorflow_probability.python.bijectors import shift
  from tensorflow_probability.python.bijectors import sigmoid
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.distributions import wishart
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.mcmc import nuts_autobatching
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import sample
label_desc: Use image node_id 16 for calling the NutsTest obj's underlying member method code with example usage: obj.testCorrelatedMVNManySteps() without return types