code: def test_batch_mix(self):
      feat = tf.random.uniform(shape=[2, 64, 5])
      desc = tf.ones(shape=[1, 64, 1])
      desc_ = tf.zeros(shape=[1, 64, 1])
      desc = tf.concat([desc, desc_], axis=0)
      (
          barycentric,
          blur_neighbours1,
          blur_neighbours2,
          indices,
      ) = permutohedral_prepare(feat)
      sliced = permutohedral_compute(
          desc,
          barycentric,
          blur_neighbours1,
          blur_neighbours2,
          indices,
          "test",
          True,
      )
      with tf.Session() as sess:
          sess.run(tf.global_variables_initializer())
          sliced_np = sess.run(sliced)
          self.assertAllClose(sliced_np[1:], np.zeros(shape=[1, 64, 1]))
  
file_path: NiftyNet/tests/crf_test.py
imports: from __future__ import absolute_import, print_function
  import numpy
  import tensorflow
  from niftynet.layer.crf import CRFAsRNNLayer
  from niftynet.layer.crf import permutohedral_prepare, permutohedral_compute
  from tests.niftynet_testcase import NiftyNetTestCase
label_desc: Use image node_id 5 for calling the CRFTest obj's underlying member method code with example usage: obj.test_batch_mix() without return types