code: def testBijectorInverse(self):
      bijector = categorical_to_discrete.CategoricalToDiscrete(
          map_values=[0.1, 0.2, 0.3, 0.4], validate_args=True
      )
      self.assertAllEqual(
          [[3, 3, 3], [0, 1, 2]],
          self.evaluate(
              bijector.inverse(
                  [[0.400001, 0.4, 0.399999], [0.1, 0.2, 0.3]]
              )
          ),
      )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.bijectors import categorical_to_discrete
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 6 for calling the CategoricalToDiscreteTest obj's underlying member method code with example usage: obj.testBijectorInverse() without return types