class_docstr: Residual block of operations, e.g. convolutional or MLP.
class_docstr_tok: ["Residual","block","of","operations",",","e.g",".","convolutional","or","MLP","."]
code: def __call__(self, x: jnp.ndarray):
      output = x
  
      # First layer in residual block.
      if self.use_layer_norm:
          output = self.layernorm1(output)
      output = self.non_linearity(output)
      output = self.inner_op1(output)
  
      # Second layer in residual block.
      if self.use_layer_norm:
          output = self.layernorm2(output)
      output = self.non_linearity(output)
      output = self.inner_op2(output)
      return x + output
  
file_path: acme/acme/jax/networks/resnet.py
imports: import enum
  import functools
  from typing import Callable, Sequence, Union
  import haiku
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling the ResidualBlock obj's underlying member method code with example usage: obj.__call__(x) and returns: unknown