code: def testBatchShape(self):
      amplitudes = np.ones((4, 1, 1, 3), np.float32)
      additive_inner_kernel = (
          exponentiated_quadratic.ExponentiatedQuadratic(
              amplitude=np.ones([2, 1, 1], np.float32),
              length_scale=np.ones([1, 3, 1], np.float32),
          )
      )
      kernel = additive_kernel.AdditiveKernel(
          kernel=additive_inner_kernel, amplitudes=amplitudes
      )
      self.assertAllEqual(tf.TensorShape([4, 2, 3]), kernel.batch_shape)
      self.assertAllEqual(
          [4, 2, 3], self.evaluate(kernel.batch_shape_tensor())
      )
  
file_path: probability/tensorflow_probability/python/experimental/psd_kernels/additive_kernel_test.py
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.psd_kernels import additive_kernel
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import exponentiated_quadratic
label_desc: Use image node_id 1 for calling the AdditiveKernelTest obj's underlying member method code with example usage: obj.testBatchShape() without return types