jaxsnn.discrete.modules.leaky_integrate_and_fire.LIFInput
-
class
jaxsnn.discrete.modules.leaky_integrate_and_fire.LIFInput(I: jax.Array, z: jax.Array) Bases:
tree_math._src.vector.VectorMixin,jaxsnn.discrete.modules.leaky_integrate_and_fire.LIFInput-
__init__(I: jax.Array, z: jax.Array) → None Initialize self. See help(type(self)) for accurate signature.
Methods
__init__(I, z)Initialize self.
asdict()astuple()tree_unflatten(_, children)Attributes
-
asdict()
-
astuple()
-
property
fields
-
tree_flatten()
-
classmethod
tree_unflatten(_, children)
-