code: def testParamTensorFromProbs(self):
      x = tf.constant([0.1, 0.5, 0.4])
      d = oc.OneHotCategorical(probs=x, validate_args=True)
      self.assertAllClose(
          *self.evaluate([tf.math.log(x), d.logits_parameter()]),
          atol=0,
          rtol=1e-4,
      )
      self.assertAllClose(
          *self.evaluate([x, d.probs_parameter()]), atol=0, rtol=1e-4
      )
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import identity
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import onehot_categorical
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 20 for calling the OneHotCategoricalTest obj's underlying member method code with example usage: obj.testParamTensorFromProbs() without return types