jaxsnn.discrete.modules.leaky_integrate.LIState

class jaxsnn.discrete.modules.leaky_integrate.LIState(V: jax.Array, I: jax.Array)

Bases: tree_math._src.vector.VectorMixin, jaxsnn.base.types.LIState

__init__(V: jax.Array, I: jax.Array) → None

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

Methods

__init__(V, I)	Initialize self.
asdict()
astuple()
tree_flatten()
tree_unflatten(_, children)

Attributes

fields

asdict()

astuple()

property fields

tree_flatten()

classmethod tree_unflatten(_, children)