spark.nn.neurons.base

Attributes

ConfigT

Classes

NeuronOutput	Generic Neuron model output spec.
NeuronConfig	Abstract Neuron model configuration class.
Neuron	Abstract Neuron model.

Module Contents

class spark.nn.neurons.base.NeuronOutput

Bases: TypedDict

Generic Neuron model output spec.

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

out_spikes: spark.core.payloads.SpikeArray

class spark.nn.neurons.base.NeuronConfig(__skip_validation__=False, **kwargs)

Bases: spark.core.config.SparkConfig

Abstract Neuron model configuration class.

Parameters:

__skip_validation__ (bool)

units: tuple[int, Ellipsis]

spark.nn.neurons.base.ConfigT

class spark.nn.neurons.base.Neuron(config=None, **kwargs)

Bases: spark.core.module.SparkModule, abc.ABC, Generic[ConfigT]

Abstract Neuron model.

This is a convenience class used to synchronize data more easily. Can be thought as the equivalent of Sequential in standard ML frameworks.

Parameters:

config (ConfigT | None)

config: ConfigT

units

reset()

Resets neuron states to their initial values.

abstractmethod __call__(in_spikes)

Execution method.

Parameters:

in_spikes (spark.core.payloads.SpikeArray)

Return type:

NeuronOutput