jaxsnn.event.types.ProbeEvent

class jaxsnn.event.types.ProbeEvent(time: jax.Array, state: StateT, probe: jax.Array)

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

__init__(time: jax.Array, state: StateT, probe: jax.Array) → None

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

Methods

__init__(time, state, probe)	Initialize self.
asdict()
astuple()
tree_flatten()
tree_unflatten(_, children)

Attributes

fields

asdict()

astuple()

property fields

tree_flatten()

classmethod tree_unflatten(_, children)