jaxsnn.event.stepping.step_existing_events.Step

class jaxsnn.event.stepping.step_existing_events.Step(parameters: Parameters, spikes: IOData, external_spikes: Optional[IOData], state: StepState, step_idx: int, layer_idx: int, queue_head: Dict[str, jax.Array], queue_indices: Dict[str, jax.Array])

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

__init__(parameters: Dict[str, jax.Array], spikes: Dict[str, jaxsnn.event.types.Spike], external_spikes: Optional[Dict[str, jaxsnn.event.types.Spike], None], state: jaxsnn.event.types.StepState, step_idx: int, layer_idx: int, queue_head: 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)