code: def testBatchShape(self):
      amplitude = np.random.uniform(2, 3.0, size=[3, 1, 2]).astype(
          np.float32
      )
      length_scale = np.random.uniform(2, 3.0, size=[1, 3, 1]).astype(
          np.float32
      )
      original_kernel = matern.GeneralizedMatern(
          df=np.pi, amplitude=amplitude, length_scale=length_scale
      )
      exponential_kernel = pointwise_exponential.PointwiseExponential(
          original_kernel
      )
      self.assertAllEqual(
          original_kernel.batch_shape, exponential_kernel.batch_shape
      )
      self.assertAllEqual(
          self.evaluate(original_kernel.batch_shape_tensor()),
          self.evaluate(exponential_kernel.batch_shape_tensor()),
      )
  
file_path: probability/tensorflow_probability/python/math/psd_kernels/pointwise_exponential_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import matern
  from tensorflow_probability.python.math.psd_kernels import parabolic
  from tensorflow_probability.python.math.psd_kernels import pointwise_exponential
label_desc: Use image node_id 2 for calling the PointwiseExponentialTest obj's underlying member method code with example usage: obj.testBatchShape() without return types