graph.LIFNeuronWithSynsRealValue
- class graph.LIFNeuronWithSynsRealValue(input_nodes: ~rockpool.graph.graph_base.SetList[GraphNode], output_nodes: ~rockpool.graph.graph_base.SetList[GraphNode], name: str, computational_module: ~typing.Any, tau_mem: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, tau_syn: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, threshold: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, bias: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, dt: float | None = None)[source]
Bases:
GenericNeurons
A
GraphModule
that encapsulates a set of LIF spiking neurons with synaptic and membrane dynamics, and with real-valued parametersAttributes overview
The time-step used for these neurons in seconds, if present
The membrane time constants of these neurons, in seconds
(Nout,)
The synaptic time constants of these neurons, in seconds
(Nin,)
The firing threshold parameters of these neurons
(Nout,)
The bias parameters of these neurons, if present
(Nout,)
The input nodes attached to this module
The output nodes attached to this module
An arbitrary name attached to this specific
GraphModule
The computational module that acts as the source for this graph module
Methods overview
__init__
(input_nodes, output_nodes, name, ...)add_input
(node)Add a
GraphNode
as an input source to this module, and connect itadd_output
(node)Add a
GraphNode
as an output of this module, and connect itRemove all
GraphNode
s as inputs of this moduleRemove all
GraphNode
s as outputs of this moduleremove_input
(node)Remove a
GraphNode
as an input of this module, and disconnect itremove_output
(node)Remove a
GraphNode
as an output of this module, and disconnect it- __init__(input_nodes: ~rockpool.graph.graph_base.SetList[GraphNode], output_nodes: ~rockpool.graph.graph_base.SetList[GraphNode], name: str, computational_module: ~typing.Any, tau_mem: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, tau_syn: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, threshold: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, bias: float | ~numpy.ndarray | ~torch.Tensor | ~jax._src.numpy.lax_numpy.array = <factory>, dt: float | None = None) None
- classmethod _convert_from(mod: GraphModuleBase) GraphModuleBase
Convert another
GraphModule
to aGraphModule
of this specific subclassYou should override this method in your subclass, to include conversion rules from other graph module classes to your specific subclass.
If you do not provide conversion rules to your specific subclass then it will not be possible to map other
GraphModule
subclasses to your subclass.- Parameters:
mod (GraphModule) – A
GraphModule
orGraphModule
subclass object to convert to an object of the specific subclass.- Returns:
A converted
GraphModule
subclass object, of the specific subclass on which this method was called.- Return type:
- classmethod _factory(size_in: int, size_out: int, name: str | None = None, computational_module: Any | None = None, *args, **kwargs) GraphModuleBase
Build a new
GraphModule
orGraphModule
subclass, with new input and outputGraphNode
s created automaticallyUse this factory method to construct a new
GraphModule
from scratch, which needs new input and outputGraphNode
s created automatically. This helper method will be inherited by newGraphModule
subclasses, and will act as factory methods also for your customGraphModule
subclass.- Parameters:
size_in (int) – The number of input
GraphNode
s to create and attachsize_out (int) – The number of output
GraphNode
s to create and attachname (str, optional) – An arbitrary name to attach to this
GraphModule
, defaults to Nonecomputational_module (Optional[Module], optional) – A rockpool computational module that forms the “generator” of this graph module, defaults to None
- Returns:
The newly constructed
GraphModule
orGraphModule
subclass- Return type:
- add_input(node: GraphNode) None
Add a
GraphNode
as an input source to this module, and connect itThe new node will be appended after the last current input node. The node will be connected with this
GraphModule
as a sink.- Parameters:
node (GraphNode) – The node to add as an input source
- add_output(node: GraphNode) None
Add a
GraphNode
as an output of this module, and connect itThe new node will be appended after the last current output node. The node will be connected with this
GraphModule
as a source.- Parameters:
node (GraphNode) – The node to add as an output
- bias: float | ndarray | Tensor | array
The bias parameters of these neurons, if present
(Nout,)
- Type:
Floatvector
- computational_module: Module
The computational module that acts as the source for this graph module
- Type:
- dt: float | None = None
The time-step used for these neurons in seconds, if present
- Type:
float
- name: str
An arbitrary name attached to this specific
GraphModule
- Type:
str
- output_nodes: SetList['GraphNode']
The output nodes attached to this module
- Type:
SetList[GraphNode]
- remove_input(node: GraphNode) None
Remove a
GraphNode
as an input of this module, and disconnect itThe node will be disconnected from this
GraphModule
as a sink, and will be removed from the module.- Parameters:
node (GraphNode) – The node to remove. If this node exists as an input to the module, it will be removed.
- remove_output(node: GraphNode) None
Remove a
GraphNode
as an output of this module, and disconnect itThe node will be disconnected from this
GraphModule
as a source, and will be removed from the module.- Parameters:
node (GraphNode) – The node to remove. If this node exists as an output to the module, it will be removed.
- tau_mem: float | ndarray | Tensor | array
The membrane time constants of these neurons, in seconds
(Nout,)
- Type:
Floatvector
- tau_syn: float | ndarray | Tensor | array
The synaptic time constants of these neurons, in seconds
(Nin,)
- Type:
Floatvector
- threshold: float | ndarray | Tensor | array
The firing threshold parameters of these neurons
(Nout,)
- Type:
Floatvector