jaxsnn.event.stepping.step_existing_events.StepState
-
class
jaxsnn.event.stepping.step_existing_events.StepState(neuron_state: StateT, time: jax.Array) Bases:
Generic[jaxsnn.base.types.StateT]-
__init__(neuron_state: StateT, time: jax.Array) → None Initialize self. See help(type(self)) for accurate signature.
Methods
__init__(neuron_state, time)Initialize self.
asdict()astuple()tree_unflatten(_, children)Attributes
-
asdict()
-
astuple()
-
property
fields
-
tree_flatten()
-
classmethod
tree_unflatten(_, children)
-