code: def test_bckm(bckm_params):
      n_features, n_instances, batch_size = bckm_params
      xshape, yshape = (n_instances[0], n_features), (
          n_instances[1],
          n_features,
      )
      np.random.seed(0)
      x = tf.convert_to_tensor(
          np.random.random(xshape).astype("float32")
      )
      y = tf.convert_to_tensor(
          np.random.random(yshape).astype("float32")
      )
  
      kernel = GaussianRBF(sigma=tf.constant(1.0))
      kernel_mat = kernel(x, y).numpy()
      bc_kernel_mat = batch_compute_kernel_matrix(
          x, y, kernel, batch_size=batch_size
      ).numpy()
      np.testing.assert_almost_equal(
          kernel_mat, bc_kernel_mat, decimal=6
      )
  
file_path: alibi-detect/alibi_detect/utils/tensorflow/tests/test_distance_tf.py
imports: import numpy
  from itertools import product
  import pytest
  import tensorflow
  from alibi_detect.utils.tensorflow import GaussianRBF, mmd2, mmd2_from_kernel_matrix, permed_lsdds
  from alibi_detect.utils.tensorflow import relative_euclidean_distance, squared_pairwise_distance
  from alibi_detect.utils.tensorflow import batch_compute_kernel_matrix
label_desc: Use image node_id 4 for calling a global function with example usage: test_bckm(bckm_params) without return types