code: def testLogProb(self):
      temperature = 0.8
      logits = [0.3, 0.1, 0.4]
      dist = gumbel_softmax.GumbelSoftmax(
          temperature, logits, validate_args=True
      )
      x = tf.constant([0, 0, 1])
      log_prob = self.evaluate(dist.log_prob(x))
      expected_log_prob = -0.972918868065
      self.assertAllClose(expected_log_prob, log_prob)
  
file_path: agents/tf_agents/distributions/gumbel_softmax_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tf_agents.distributions import gumbel_softmax
label_desc: Use image node_id 1 for calling the GumbelSoftmaxTest obj's underlying member method code with example usage: obj.testLogProb() without return types