class_docstr: Categorical distribution for discrete action spaces.
docstr: Args:
      logits: A float32 tensor with shape [BATCH_SIZE, NUM_ACTIONS] of unnormalized policy logits
docstr_tok: ["Args",":","logits",":","A","float32","tensor","with","shape","[","BATCH_SIZE",",","NUM_ACTIONS","]","of","unnormalized","policy","logits"]
code: def __init__(self, logits):
      """
      Args:
          logits: A float32 tensor with shape [BATCH_SIZE, NUM_ACTIONS] of unnormalized policy logits
      """
      assert len(logits.shape) == 2
      self.logits = logits
  
imports: from parl.core.fluid import layers
label_desc: Use image node_id 1 to create a new CategoricalDistribution object from inherited base classes: PolicyDistribution with example: obj = CategoricalDistribution(logits)