code: def test_batch_shape_tensor(self, is_static, get_distributions):
      _, concat_dist = get_distributions(is_static=is_static)
      self.assertAllEqual(
          self.evaluate(concat_dist.batch_shape_tensor()), [2, 6, 4]
      )
  
imports: from absl.testing import parameterized
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import batch_concat
  from tensorflow_probability.python.distributions import dirichlet
  from tensorflow_probability.python.distributions import exponential
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import logistic
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import onehot_categorical
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the BatchConcatTest obj's underlying member method code with example usage: obj.test_batch_shape_tensor(is_static, get_distributions) without return types