jaxsnn.event.types.ProbeStepState

class jaxsnn.event.types.ProbeStepState(neuron_state: StateT, time: float, input_queue: jax.Array, probe_queue: jax.Array)

Bases: Generic[jaxsnn.base.types.StateT]

__init__(neuron_state: StateT, time: float, input_queue: jax.Array, probe_queue: jax.Array) → None

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

Methods

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

Attributes

fields

asdict()

astuple()

property fields

tree_flatten()

classmethod tree_unflatten(_, children)