jaxsnn.discrete.leaky_integrate.LIState

class jaxsnn.discrete.leaky_integrate.LIState(v: jax.Array, i: jax.Array)

Bases: tree_math._src.vector.VectorMixin, jaxsnn.discrete.leaky_integrate.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)