class_docstr: Thin wrapper around Deterministic that supports taking log_prob.
class_docstr_tok: ["Thin","wrapper","around","Deterministic","that","supports","taking","log_prob","."]
docstr: Takes log-probs by casting to tf.float32 instead of self.dtype.
docstr_tok: ["Takes","log-probs","by","casting","to","tf.float32","instead","of","self.dtype","."]
code: def _log_prob(self, x):
      """Takes log-probs by casting to tf.float32 instead of self.dtype."""
      return tf.math.log(tf.cast(self.prob(x), dtype=tf.float32))
  
file_path: agents/tf_agents/policies/greedy_policy.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Optional, Text
  import gin
  import tensorflow
  import tensorflow_probability
  from tf_agents.policies import tf_policy
  from tf_agents.trajectories import policy_step
label_desc: Use image node_id 1 for calling the DeterministicWithLogProb obj's underlying member method code with example usage: obj._log_prob(x) and returns: tf