class_docstr: Module to compose parameterized functions, implemented as a scan.
class_docstr_tok: ["Module","to","compose","parameterized","functions",",","implemented","as","a","scan","."]
docstr: Iterate f count times, with non-shared parameters.
docstr_tok: ["Iterate","f","count","times",",","with","non-shared","parameters","."]
code: def __init__(
      self,
      count: int,
      unroll: int,
      pass_reverse_to_layer_fn: bool = False,
      transparency_map: Optional[LayerStackTransparencyMapping] = None,
      name: str = "",
  ):
      """Iterate f count times, with non-shared parameters."""
      self._name = name
      self._count = count
      self._unroll = unroll
      self._pass_reverse_to_layer_fn = pass_reverse_to_layer_fn
      self._transparency_map = transparency_map
  
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 _LayerStack object  with example: obj = _LayerStack(count, unroll, pass_reverse_to_layer_fn, transparency_map, name)