jaxsnn.event.stepping.step_multi_layer.Carry

class jaxsnn.event.stepping.step_multi_layer.Carry(parameters: Parameters, spikes: IOData, external_spikes: Optional[IOData], states: States, queue_heads: Dict[str, jax.Array], queue_indices: Dict[str, jax.Array])

Bases: tree_math._src.vector.VectorMixin, jaxsnn.event.types.Carry

__init__(parameters: Dict[str, jax.Array], spikes: Dict[str, jaxsnn.event.types.Spike], external_spikes: Optional[Dict[str, jaxsnn.event.types.Spike], None], states: Dict[str, StateT], queue_heads: Dict[str, jax.Array], queue_indices: Dict[str, jax.Array]) → None

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(parameters, spikes, …)	Initialize self.
asdict()
astuple()
tree_flatten()
tree_unflatten(_, children)

Attributes

fields

asdict()

astuple()

property fields

tree_flatten()

classmethod tree_unflatten(_, children)