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","."]
docstr: Returns a (Logits, Value) tuple.
docstr_tok: ["Returns","a","(","Logits",",","Value",")","tuple","."]
code: def __call__(
      self, inputs: jnp.ndarray
  ) -> Tuple[jnp.ndarray, jnp.ndarray]:
      """Returns a (Logits, Value) tuple."""
      logits = self._policy_layer(inputs)  # [B, A]
      value = jnp.squeeze(self._value_layer(inputs), axis=-1)  # [B]
  
      return logits, value
  
imports: from typing import Tuple
  import haiku
  import jax.numpy
label_desc: Use image node_id 2 for calling the PolicyValueHead obj's underlying member method code with example usage: obj.__call__(inputs) and returns: logits,  value