class_docstr: _LayerStack impl with per-layer inputs provided to the function.
class_docstr_tok: ["_LayerStack","impl","with","per-layer","inputs","provided","to","the","function","."]
code: def __init__(
      self,
      f: WrappedFn,
      count: int,
      unroll: int,
      pass_reverse_to_layer_fn: bool = False,
      transparency_map: Optional[LayerStackTransparencyMapping] = None,
      name: str = "",
  ):
      super().__init__(
          count=count,
          unroll=unroll,
          pass_reverse_to_layer_fn=pass_reverse_to_layer_fn,
          transparency_map=transparency_map,
          name=name,
      )
      self._f = f
  
imports: import collections
  import functools
  import inspect
  from typing import Any, Callable, Optional, Protocol, Union
  from haiku._src import base
  from haiku._src import lift
  from haiku._src import module
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 1 to create a new _LayerStackWithPerLayer object from inherited base classes: _LayerStack with example: obj = _LayerStackWithPerLayer(f, count, unroll, pass_reverse_to_layer_fn, transparency_map, name)