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
      )
      scale_mixture_rate = np.random.uniform(
          2, 3.0, size=[3, 1, 1]
      ).astype(np.float32)
      k = rational_quadratic.RationalQuadratic(
          amplitude, length_scale, scale_mixture_rate=scale_mixture_rate
      )
      self.assertAllEqual(tf.TensorShape([3, 3, 2]), k.batch_shape)
      self.assertAllEqual(
          [3, 3, 2], self.evaluate(k.batch_shape_tensor())
      )
  
file_path: probability/tensorflow_probability/python/math/psd_kernels/rational_quadratic_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import rational_quadratic
label_desc: Use image node_id 3 for calling the RationalQuadraticTest obj's underlying member method code with example usage: obj.testBatchShape() without return types