code: def testBatchShape(self):
      base_kernel = matern.MaternFiveHalves()
      isd_continuous = np.ones([3, 1, 2], dtype=np.float32)
      isd_categorical = np.ones([1, 4, 1], dtype=np.float32)
      kernel = fswc.FeatureScaledWithCategorical(
          base_kernel,
          inverse_scale_diag=fswc.ContinuousAndCategoricalValues(
              isd_continuous, isd_categorical
          ),
          validate_args=True,
      )
      self.assertAllEqual(tf.TensorShape([3, 4]), kernel.batch_shape)
      self.assertAllEqual(
          [3, 4], self.evaluate(kernel.batch_shape_tensor())
      )
  
file_path: probability/tensorflow_probability/python/experimental/psd_kernels/feature_scaled_with_categorical_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.psd_kernels import feature_scaled_with_categorical
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import exponentiated_quadratic
  from tensorflow_probability.python.math.psd_kernels import feature_scaled
  from tensorflow_probability.python.math.psd_kernels import matern
label_desc: Use image node_id 1 for calling the FeatureScaledWithCategoricalTest obj's underlying member method code with example usage: obj.testBatchShape() without return types