jaxsnn.event.transition.StepState

class jaxsnn.event.transition.StepState(neuron_state: State, spike_times: jax.Array, spike_mask: jax.Array, time: float, input_queue: InputQueue)

Bases: Generic[jaxsnn.event.types.State]

__init__(neuron_state: State, spike_times: jax.Array, spike_mask: jax.Array, time: float, input_queue: jaxsnn.event.types.InputQueue) → None

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

Methods

__init__(neuron_state, spike_times, …)	Initialize self.
asdict()
astuple()
tree_flatten()
tree_unflatten(_, children)

Attributes

fields

asdict()

astuple()

property fields

tree_flatten()

classmethod tree_unflatten(_, children)