class_docstr: A network with two linear layers, for policy and value respectively.
class_docstr_tok: ["A","network","with","two","linear","layers",",","for","policy","and","value","respectively","."]
code: def __init__(self, num_actions: int):
      super().__init__(name="policy_value_network")
      self._policy_layer = hk.Linear(num_actions)
      self._value_layer = hk.Linear(1)
  
imports: from typing import Tuple
  import haiku
  import jax.numpy
label_desc: Use image node_id 1 to create a new PolicyValueHead object from inherited base classes: hk with example: obj = PolicyValueHead(num_actions)