API reference¶
Main¶
|
NengoLoihi simulator for Loihi hardware and emulator. |
|
Add nengo_loihi config option to network. |
|
Specifies how an ensemble should be split across Loihi neuron cores. |
Modify Nengo’s default parameters for better performance with Loihi. |
-
class
nengo_loihi.
Simulator
(network, dt=0.001, seed=None, model=None, precompute=None, target=None, progress_bar=None, remove_passthrough=True, hardware_options=None)[source]¶ NengoLoihi simulator for Loihi hardware and emulator.
The simulator takes a
nengo.Network
and builds internal data structures to run the model defined by that network on Loihi emulator or hardware. Run the simulator with theSimulator.run
method, and access probed data through thedata
attribute.Building and running the simulation allocates resources. To properly free these resources, call the
Simulator.close
method. Alternatively,Simulator.close
will automatically be called if you usewith
syntax:with nengo_loihi.Simulator(my_network) as sim: sim.run(0.1) print(sim.data[my_probe])
Note that the
data
attribute is still accessible even when a simulator has been closed. Running the simulator, however, will raise an error.- Parameters
- networkNetwork
A network object to be built and then simulated. If None, then the model parameter must be provided instead.
- dtfloat, optional (Default: 0.001)
The length of a simulator timestep, in seconds.
- seedint, optional (Default: None)
A seed for all stochastic operators used in this simulator. Will be set to
network.seed + 1
if not given.- modelModel, optional (Default: None)
A
Model
that contains build artifacts to be simulated. Usually the simulator will build this model for you; however, if you want to build the network manually, or you want to inject build artifacts in the model before building the network, then you can pass in aModel
instance.- precomputebool, optional (Default: None)
Whether model inputs should be precomputed to speed up simulation. When precompute is False, the simulator will be run one step at a time in order to use model outputs as inputs in other parts of the model. By default, the simulator will choose
True
if it works for your model, andFalse
otherwise.- targetstr, optional (Default: None)
Whether the simulator should target the emulator (
'sim'
) or Loihi hardware ('loihi'
). If None, target will default to'loihi'
if NxSDK is installed, and the emulator if it is not.- progress_barbool or
nengo.utils.progress.ProgressBar
, optional Progress bar for displaying build and simulation progress. If
True
, the default progress bar will be used. IfFalse
, the progress bar will be disabled. For more control, pass in aProgressBar
instance.- remove_passthroughbool, optional
Whether to remove passthrough
nengo.Node
objects from the model (i.e. Nodes withoutput=None
that only act as intermediaries between other objects). This will often allow more of the network to be run on-chip, but can sometimes require significantly larger connections (e.g. more input and output axons).- hardware_optionsdict, optional (Default: {})
Dictionary of additional configuration for the hardware. See
hardware.HardwareInterface
for possible parameters.
- Attributes
- closedbool
Whether the simulator has been closed. Once closed, it cannot be reopened.
- dataProbeDict
The dictionary mapping from Nengo objects to the data associated with those objects. In particular, each
nengo.Probe
maps to the data probed while running the simulation.- modelModel
The
Model
containing the data structures necessary for simulating the network.- precomputebool
Whether model inputs should be precomputed to speed up simulation. When precompute is False, the simulator will be run one step at a time in order to use model outputs as inputs in other parts of the model.
-
property
dt
¶ (float) The step time of the simulator.
-
property
n_steps
¶ (int) The current time step of the simulator.
-
property
time
¶ (float) The current time of the simulator.
-
close
()[source]¶ Closes the simulator.
Any call to
Simulator.run
,Simulator.run_steps
,Simulator.step
, andSimulator.reset
on a closed simulator raises aSimulatorClosed
exception.
-
reset
(seed=None)[source]¶ Reset the simulator state.
- Parameters
- seedint, optional
A seed for all stochastic operators used in the simulator. This will change the random sequences generated for noise or inputs (e.g. from processes), but not the built objects (e.g. ensembles, connections).
-
run
(time_in_seconds)[source]¶ Simulate for the given length of time.
If the given length of time is not a multiple of
dt
, it will be rounded to the nearestdt
. For example, ifdt
is 0.001 andrun
is called withtime_in_seconds=0.0006
, the simulator will advance one timestep, resulting in the actual simulator time being 0.001.The given length of time must be positive. The simulator cannot be run backwards.
- Parameters
- time_in_secondsfloat
Amount of time to run the simulation for. Must be positive.
-
run_steps
(steps)[source]¶ Simulate for the given number of
dt
steps.- Parameters
- stepsint
Number of steps to run the simulation for.
-
trange
(sample_every=None, dt=None)[source]¶ Create a vector of times matching probed data.
Note that the range does not start at 0 as one might expect, but at the first timestep (i.e.,
dt
).- Parameters
- sample_everyfloat, optional (Default: None)
The sampling period of the probe to create a range for. If None, a time value for every
dt
will be produced.
-
nengo_loihi.
add_params
(network)[source]¶ Add nengo_loihi config option to network.
The following options will be added:
nengo.Ensemble
on_chip
: Whether the ensemble should be simulated on a Loihi chip. Marking specific ensembles for simulation off of a Loihi chip can help with debugging.block_shape
: Specifies how this ensemble should be split across Loihi neuron cores. SeeBlockShape
for more details.
nengo.Connection
pop_type
: The axon format when using population spikes, which are only used for convolutional connections. Must be either the integer 16 or 32. By default, we usepop_type
32. Using 16 reduces the number of axons required by a factor of two, but has more constraints, including on the number of channels allowed per block. When usingpop_type
16, we recommend to usechannels_last=True
and haven_filters
(as well as the number of channels per block) be a multiple of 4 on your convolutional connections; this will reduce the required synapse memory.
Examples
>>> with nengo.Network() as model: ... ens = nengo.Ensemble(10, dimensions=1) ... # By default, ens will be placed on a Loihi chip ... nengo_loihi.add_params(model) ... model.config[ens].on_chip = False ... # Now it will be simulated with Nengo
-
class
nengo_loihi.
BlockShape
(shape, ensemble_shape_or_transform)[source]¶ Specifies how an ensemble should be split across Loihi neuron cores.
Each neuron core can have, at most, 1024 compartments. Ensembles with more than 1024 neurons are automatically split such that they will be distributed evenly across as few cores as possible.
This class allows you to split a smaller ensemble onto multiple cores. It also allows you to control the splitting process for any ensemble, which is particularly useful when that ensemble participates in a convolutional connection.
- Parameters
- shapetuple of int
The conceptual shape of the compartments on each core after splitting. This tuple must have the same number of elements as
ensemble_shape
.- ensemble_shape_or_transformtuple of int or
nengo.Convolution
The conceptual shape of the neurons in the ensemble. If a
nengo.Convolution
instance is passed, the conceptual shape is inferred automatically.
Examples
Split an ensemble across two Loihi blocks.
with nengo.Network() as net: nengo_loihi.add_params(net) ens = nengo.Ensemble(10, 1) net.config[ens].block_shape = nengo_loihi.BlockShape((5,), (10,)) print(net.config[ens].block_shape.n_splits)
2
Interpret an ensemble as a two dimensional layer and split into four blocks.
with nengo.Network() as net: nengo_loihi.add_params(net) ens = nengo.Ensemble(16, 1) net.config[ens].block_shape = nengo_loihi.BlockShape((2, 2), (4, 4)) print(net.config[ens].block_shape.n_splits)
4
- Attributes
- shapetuple of int
The conceptual shape of the compartments on each core after splitting.
- ensemble_shapetuple of int
The conceptual shape of the neurons in the ensemble.
-
nengo_loihi.
set_defaults
()[source]¶ Modify Nengo’s default parameters for better performance with Loihi.
The following defaults will be modified:
nengo.Ensemble
max_rates
: Set toUniform(low=100, high=120)
intercepts
: Set toUniform(low=-0.5, high=0.5)
nengo.LIF
andnengo.SpikingRectifiedLinear
initial_voltage
: Set to 0
Neurons¶
|
Simulate LIF neurons as done by Loihi. |
Simulate spiking rectified linear neurons as done by Loihi. |
-
class
nengo_loihi.neurons.
LoihiLIF
(tau_rc=0.02, tau_ref=0.002, min_voltage=0, amplitude=1, nengo_dl_noise=None, **kwargs)[source]¶ Simulate LIF neurons as done by Loihi.
On Loihi, the inter-spike interval has to be an integer. This causes aliasing the firing rates where a wide variety of inputs can produce the same output firing rate. This class reproduces this effect, as well as the discretization of some of the neuron parameters. It can be used in e.g.
nengo
ornengo_dl
to reproduce these unique Loihi effects.- Parameters
- nengo_dl_noise
nengo_extras.loihi_training.NeuronOutputNoise
Noise added to the rate-neuron output when training with this neuron type in
nengo_dl
.
- nengo_dl_noise
-
class
nengo_loihi.neurons.
LoihiSpikingRectifiedLinear
(amplitude=1, **kwargs)[source]¶ Simulate spiking rectified linear neurons as done by Loihi.
On Loihi, the inter-spike interval has to be an integer. This causes aliasing in the firing rates such that a wide variety of inputs produce the same output firing rate. This class reproduces this effect. It can be used in e.g.
nengo
ornengo_dl
to reproduce these unique Loihi effects.
Decode neurons¶
Decode neurons facilitate NEF-style connections on the chip.
The type of decode neurons used by a model can be set on builder.Model
.
Defines parameters for a group of decode neurons. |
|
One or more pairs of on/off neurons per dimension. |
|
Uses multiple on/off neuron pairs per dimension, plus noise. |
|
Uses five heterogeneous on/off pairs with pre-set values per dimension. |
|
Uses ten heterogeneous on/off pairs with pre-set values per dimension. |
-
class
nengo_loihi.decode_neurons.
DecodeNeurons
(dt=0.001)[source]¶ Defines parameters for a group of decode neurons.
DecodeNeurons are used on the chip to facilitate NEF-style connections, where activities from a neural ensemble are first transformed into a decoded value (which is stored in the activities and synapses of the spiking decode neurons), before being passed on to another ensemble (via that ensemble’s encoders).
- Parameters
- dtfloat
Time step used by the simulator.
-
get_block
(weights, block_label=None, syn_label=None)[source]¶ Get a LoihiBlock for implementing neurons on the chip.
- Parameters
- weights(n, d) ndarray
Weights that project the
n
inputs to thed
dimensions represented by these neurons. Typically, the inputs will be neurons belonging to an Ensemble, and these weights will be decoders.- block_labelstring (Default: None)
Optional label for the LoihiBlock.
- syn_labelstring (Default: None)
Optional label for the Synapse.
- Returns
- blockLoihiBlock
The neurons on the chip.
- synSynapse
The synapses connecting into the chip neurons.
-
get_ensemble
(dim, add_to_container=True)[source]¶ Get a Nengo Ensemble for implementing neurons on the host.
- Parameters
- dimint
Number of dimensions to be represented by these neurons.
- add_to_containerbool, optional (Default: True)
Whether to add the ensemble to the currently active network.
- Returns
- ensEnsemble
An Ensemble for implementing these neurons in a Nengo network.
-
get_post_encoders
(encoders)[source]¶ Encoders for post population that these neurons connect in to.
- Parameters
- encoders(n, d) ndarray
Regular scaled encoders for the ensemble, which map the ensemble’s
d
input dimensions to itsn
neurons.
- Returns
- decode_neuron_encoders(?, n) ndarray
Encoders for mapping these neurons to the post-ensemble’s neurons. The number of rows depends on how
get_post_inds
is being used (i.e. there could be one row per neuron in this block, or there could be fewer rows withget_post_inds
mapping multiple neurons to each row).
-
get_post_inds
(inds, d)[source]¶ Indices for mapping neurons to post-encoder dimensions.
- Parameters
- indslist of ints
Indices for mapping decode neuron dimensions to post-ensemble dimensions. Usually, this will be determined by a slice on the post ensemble in a connection (which maps the output of the transform/function to select dimensions on the post ensemble).
- dint
Number of dimensions in the post-ensemble.
-
class
nengo_loihi.decode_neurons.
OnOffDecodeNeurons
(pairs_per_dim=1, dt=0.001, rate=None, is_input=False)[source]¶ One or more pairs of on/off neurons per dimension.
In this class itself, all the pairs in a dimension are identical. It can still be advantageous to have more than one pair per dimension, though, since this can allow all neurons to have lower firing rates and thus act more linearly (due to period aliasing at high firing rates). Subclasses may use pairs that are not identical (by adding noise or heterogeneity).
- Parameters
- pairs_per_dimint
Number of repeated neurons per dimension. Currently, all DecodeNeuron classes use separate on/off neuron pairs for each dimension. This is the number of such pairs per dimension.
- dtfloat
Time step used by the simulator.
- ratefloat (Default: None)
Max firing rate of each neuron. By default, this is chosen so that the sum of all repeated neuron rates is
1. / dt
, and thus as a group the neurons average one spike per timestep.- is_inputbool (Default: False)
Whether these decode neurons are being used to provide input.
-
get_block
(weights, block_label=None, syn_label=None)[source]¶ Get a LoihiBlock for implementing neurons on the chip.
- Parameters
- weights(n, d) ndarray
Weights that project the
n
inputs to thed
dimensions represented by these neurons. Typically, the inputs will be neurons belonging to an Ensemble, and these weights will be decoders.- block_labelstring (Default: None)
Optional label for the LoihiBlock.
- syn_labelstring (Default: None)
Optional label for the Synapse.
- Returns
- blockLoihiBlock
The neurons on the chip.
- synSynapse
The synapses connecting into the chip neurons.
-
get_ensemble
(dim, add_to_container=True)[source]¶ Get a Nengo Ensemble for implementing neurons on the host.
- Parameters
- dimint
Number of dimensions to be represented by these neurons.
- add_to_containerbool, optional (Default: True)
Whether to add the ensemble to the currently active network.
- Returns
- ensEnsemble
An Ensemble for implementing these neurons in a Nengo network.
-
get_post_encoders
(encoders)[source]¶ Encoders for post population that these neurons connect in to.
- Parameters
- encoders(n, d) ndarray
Regular scaled encoders for the ensemble, which map the ensemble’s
d
input dimensions to itsn
neurons.
- Returns
- decode_neuron_encoders(?, n) ndarray
Encoders for mapping these neurons to the post-ensemble’s neurons. The number of rows depends on how
get_post_inds
is being used (i.e. there could be one row per neuron in this block, or there could be fewer rows withget_post_inds
mapping multiple neurons to each row).
-
get_post_inds
(inds, d)[source]¶ Indices for mapping neurons to post-encoder dimensions.
- Parameters
- indslist of ints
Indices for mapping decode neuron dimensions to post-ensemble dimensions. Usually, this will be determined by a slice on the post ensemble in a connection (which maps the output of the transform/function to select dimensions on the post ensemble).
- dint
Number of dimensions in the post-ensemble.
-
class
nengo_loihi.decode_neurons.
NoisyDecodeNeurons
(pairs_per_dim, dt=0.001, rate=None, noise_exp=- 2.0)[source]¶ Uses multiple on/off neuron pairs per dimension, plus noise.
The noise allows each on-off neuron pair to do something different. The population average is a better representation of the encoded value than can be achieved with a single on/off neuron pair (if the magnitude of the noise is correctly calibrated).
- Parameters
- pairs_per_dimint
Number of repeated neurons per dimension. Currently, all DecodeNeuron classes use separate on/off neuron pairs for each dimension. This is the number of such pairs per dimension.
- dtfloat
Time step used by the simulator.
- ratefloat (Default: None)
Max firing rate of each neuron. By default, this is chosen so that the sum of all repeated neuron rates is
1. / dt
, and thus as a group the neurons average one spike per timestep.- noise_expfloat, optional (Default: -2.)
Base-10 exponent for noise added to neuron voltages.
-
get_block
(weights, block_label=None, syn_label=None)[source]¶ Get a LoihiBlock for implementing neurons on the chip.
- Parameters
- weights(n, d) ndarray
Weights that project the
n
inputs to thed
dimensions represented by these neurons. Typically, the inputs will be neurons belonging to an Ensemble, and these weights will be decoders.- block_labelstring (Default: None)
Optional label for the LoihiBlock.
- syn_labelstring (Default: None)
Optional label for the Synapse.
- Returns
- blockLoihiBlock
The neurons on the chip.
- synSynapse
The synapses connecting into the chip neurons.
Builder¶
The builder turns a Nengo network into a builder.Model
appropriate for running on Loihi.
This model is mainly composed of LoihiBlock
objects,
which map parts of the network to Loihi compartments, axons, and synapses.
The model also contains LoihiInput
objects to provide input,
and LoihiProbe
objects to collect output.
|
The data structure for the emulator/hardware simulator. |
Fills in the Loihi Model object based on the Nengo Network. |
|
|
Class holding Loihi objects that can be placed on the chip. |
|
Stores information for configuring Loihi compartments. |
|
A group of axons targeting a specific Synapse object. |
|
A group of Loihi synapses that share some properties. |
|
|
|
|
|
Record data from one or more LoihiBlock target states. |
-
class
nengo_loihi.builder.
Model
(dt=0.001, label=None, builder=None)[source]¶ The data structure for the emulator/hardware simulator.
Defines methods for adding inputs, blocks and probes. Also handles build functions, and information associated with building the Nengo model.
Some of the Model attributes can be modified before the build process to change how certain build functions work. Those attributes are listed first in the attributes section below. To change them, create an instance of
Model
and pass it in toSimulator
along with your network:model = nengo_loihi.builder.Model(dt=0.001) model.pes_error_scale = 50. with nengo_loihi.Simulator(network, model=model) as sim: sim.run(1.0)
- Parameters
- dtfloat, optional (Default: 0.001)
The length of a simulator timestep, in seconds.
- labelstr, optional (Default: None)
A name or description to differentiate models.
- builderBuilder, optional (Default: None)
A
Builder
instance to keep track of build functions. If None, the default builder will be used.
- Attributes
- Build parameters
- decode_neuronsDecodeNeurons
Type of neurons used to facilitate decoded (NEF-style) connections.
- decode_taufloat
Time constant of lowpass synaptic filter used with decode neurons.
- intercept_limitfloat
Limit for clipping intercepts, to avoid neurons with high gains.
- node_neuronsDecodeNeurons
Type of neurons used to convert real-valued node outputs to spikes for the chip.
- pes_error_scalefloat
Scaling for PES errors, before rounding and clipping to -127..127.
- pes_wgt_expint
Learning weight exponent (base 2) for PES learning connections. This controls the maximum weight magnitude (where a larger exponent means larger potential weights, but lower weight resolution).
- validation_levelValidationLevel
Level of validation to do during the build process. Defaults to ValidationLevel.MINIMAL.
- vth_nonspikingfloat
Voltage threshold for non-spiking neurons (i.e. voltage decoders).
- Internal attributes
- blocksdict
Mapping from Loihi blocks to a unique integer for that block.
- block_shapesdict
Mapping from Loihi blocks to
BlockShape
instances.- builderBuilder
The build functions used by this model.
- connection_decode_neuronsdict of {Connection: Ensemble}
Map of each
nengo.Connection
that requires DecodeNeurons, to thenengo.Ensemble
that implements said DecodeNeurons.- chip2host_paramsdict
Mapping from Nengo objects to any additional parameters associated with those objects for use during the build process.
- dtfloat
The length of a simulator timestep, in seconds.
- inputslist of LoihiInput
List of inputs to this model.
- labelstr or None
A name or description to differentiate models.
- objsdict
Dictionary mapping from Nengo objects to NengoLoihi objects.
- paramsdict
Mapping from objects to namedtuples containing parameters generated in the build process.
- probeslist
List of probes on this model.
- seededdict
All objects are assigned a seed, whether the user defined the seed or it was automatically generated. ‘seeded’ keeps track of whether the seed is user-defined. We consider the seed to be user-defined if it was set directly on the object, or if a seed was set on the network in which the object resides, or if a seed was set on any ancestor network of the network in which the object resides.
- seedsdict
Mapping from objects to the integer seed assigned to that object.
-
utilization_summary
()[source]¶ Summarize utilization info for all blocks.
- Returns
- summarylist of string
One string per block, with the block name and the block utilization values, expressed as percentages.
Examples
with nengo.Network() as network: nengo.Ensemble(1000, 3, label="MyEnsemble") with nengo_loihi.Simulator(network) as sim: print("\n".join(sim.model.utilization_summary()))
LoihiBlock(<Ensemble "MyEnsemble">): 97.7% compartments, 0.0% in-axons, 0.0% out-axons, 0.0% synapses Average (1 blocks): 97.7% compartments, 0.0% in-axons, 0.0% out-axons, 0.0% synapses
-
class
nengo_loihi.builder.
Builder
[source]¶ Fills in the Loihi Model object based on the Nengo Network.
We cannot use the Nengo builder as is because we make normal Nengo networks for host-to-chip and chip-to-host communication. To keep Nengo and NengoLoihi builders separate, we make a subclass.
-
class
nengo_loihi.block.
LoihiBlock
(n_neurons, label=None)[source]¶ Class holding Loihi objects that can be placed on the chip.
This class can be thought of as a block of the Loihi board, and is how NengoLoihi keeps track of how Loihi Neuron cores will be configured. Generally, the job of the build process is to convert Nengo objects (ensembles, connections, and nodes) to LoihiBlocks, which will then be used by the
EmulatorInterface
orHardwareInterface
.Initially, parameters in a LoihiBlock are floating point values. The
discretize_block
function converts them to integer values inplace for use on Loihi.- Attributes
- axonslist of Axon
Axon objects outputting from the compartments in the block.
- compartmentCompartment
Compartment object representing all compartments in the block.
- n_neuronsint
The number of neurons in the block.
- named_synapsesdict {str: Synape}
Mapping from a name to a Synapse object.
- labelstring
A label for the block (for debugging purposes).
- synapseslist of Synapse
Synapse objects projecting to compartments in the block.
-
utilization
()[source]¶ Measure utilization of core resources by the block.
- Returns
- utilizationOrderedDict
“compartments”: The (current, max) number of compartments used. “in-axons”: The (current, max) number of input axons used. “out-axons”: The (current, max) number of output axons used. “synapses”: The (current, max) amount of synapse memory used.
-
class
nengo_loihi.block.
Compartment
(n_compartments, label=None)[source]¶ Stores information for configuring Loihi compartments.
The information stored here will be associated with some block, and all compartments will share certain information. While compartments are usually thought of neurons, we use compartments to implement Nengo ensembles, nodes, and connection through special decode neurons.
Before
discretize_compartment
has been called, most attributes in this class are floating-point values. Callingdiscretize_compartment
converts them to integer values inplace for use on Loihi.- Attributes
- bias(n,) ndarray
Compartment biases.
- enable_noise(n,) ndarray
Whether to enable noise for each compartment.
- decay_u(n,) ndarray
Input (synapse) decay constant for each compartment.
- decay_v(n,) ndarray
Voltage decay constant for each compartment.
- labelstring
A label for the block (for debugging purposes).
- n_compartmentsint
The number of compartments in the block.
- noise_at_membrane{0, 1}
Inject noise into current (0) or voltage (1).
- noise_expfloat or int
Exponent for noise generation. Floating point values are base 10 in units of current or voltage. Integer values are in base 2.
- noise_offsetfloat or int
Offset for noise generation.
- refract_delay(n,) ndarray
Compartment refractory delays, in time steps.
- scale_ubool
Scale input (U) by decay_u so that the integral of U is the same before and after filtering.
- scale_vbool
Scale voltage (V) by decay_v so that the integral of V is the same before and after filtering.
- tau_sfloat or None
Time constant used to set decay_u. None if decay_u has not been set.
- vmaxfloat or int (range [2**9 - 1, 2**23 - 1])
Maximum voltage for all compartments, in the same units as
vth
.- vminfloat or int (range [-2**23 + 1, 0])
Minimum voltage for all compartments, in the same units as
vth
.- vth(n,) ndarray
Compartment voltage thresholds.
-
configure_default_filter
(tau_s, dt=0.001)[source]¶ Set the default Lowpass synaptic input filter for compartments.
- Parameters
- tau_sfloat
nengo.Lowpass
synapse time constant for filtering.- dtfloat
Simulator time step.
-
configure_filter
(tau_s, dt=0.001)[source]¶ Set Lowpass synaptic input filter for compartments.
- Parameters
- tau_sfloat
nengo.Lowpass
synapse time constant for filtering.- dtfloat
Simulator time step.
-
configure_lif
(tau_rc=0.02, tau_ref=0.001, vth=1, dt=0.001, min_voltage=0)[source]¶ Configure these compartments as individual LIF neurons.
- Parameters
- tau_rcfloat
Membrane time constant (in seconds) of the neurons.
- tau_reffloat
Refractory period (in seconds) of the neurons.
- vthfloat
Voltage firing threshold of the neurons.
- dtfloat
Simulator time step length (in seconds).
- min_voltagefloat
The minimum voltage for the neurons.
-
configure_nonspiking
(tau_ref=0.0, vth=1, dt=0.001)[source]¶ Configure these compartments as individual non-spiking neurons.
- Parameters
- tau_reffloat
Refractory period (in seconds) of the neurons.
- vthfloat
Voltage firing threshold of the neurons.
- dtfloat
Simulator time step length (in seconds).
-
configure_relu
(tau_ref=0.0, vth=1, dt=0.001)[source]¶ Configure these compartments as individual Rectified Linear neurons.
These are also known as non-leaky integrate-and-fire neurons. The voltage is the integral of the input current.
- Parameters
- tau_reffloat
Refractory period (in seconds) of the neurons.
- vthfloat
Voltage firing threshold of the neurons.
- dtfloat
Simulator time step length (in seconds).
-
class
nengo_loihi.block.
Axon
(n_axons, label=None)[source]¶ A group of axons targeting a specific Synapse object.
- Attributes
- compartment_atomslist of length
block.n_neurons
Atom (weight index) associated with each block compartment.
- compartment_maplist of length
block.n_neurons
Index of the axon in
target
targeted by each block compartment.- n_axonsint
The number of outgoing axons.
- targetSynapse
Target synapses for these axons.
- compartment_atomslist of length
-
class
Spike
(axon_idx, atom=0)[source]¶ A spike targeting a particular axon within a Synapse object.
The Synapse target is implicit, given by the Axon object that creates this Spike.
- Parameters
- axon_idxint
The index of the axon within the targeted Synapse object.
- atomint, optional (Default: 0)
An index into the target Synapse weights. This allows spikes targeting a particular axon to use different weights.
-
property
slots_per_axon
¶ The number of axon_cfg slots occupied by each axon.
-
set_compartment_axon_map
(target_axons, atoms=None)[source]¶ Set mapping from compartments to axons in target.
- Parameters
- target_axonsarray_like (
n_compartments
,) Indices indicating which target axon each compartment maps to. If < 0, the corresponding compartment will not be used with these axons.
- atomsarray_like (
n_compartments
,) Atoms to use for each compartment. Use only if
pop_type != 0
.
- target_axonsarray_like (
-
class
nengo_loihi.block.
Synapse
(n_axons, label=None)[source]¶ A group of Loihi synapses that share some properties.
- Attributes
- axon_compartment_baseslist or None
List providing base (compartment offset) for each input axon.
- axon_to_weight_mapdict or None
Map from input axon index to weight index, to allow weights to be re-used by axons. If None, the weight index for an input axon is the axon index.
- indices(population, axon, compartment) ndarray
The synapse indices.
- learningbool
Whether synaptic tracing and learning is enabled for these synapses.
- learning_ratefloat
The learning rate.
- learning_wgt_expint
The weight exponent used on this connection if learning is enabled.
- n_axonsint
Number of input axons to this group of synapses.
- pop_typeint (0, 16, 32)
Whether these synapses are discrete (0), pop16, or pop32. This determines the type of axons these synapses can connect to.
- synapse_cfgSynapseConfig
The synapse format object for these synapses.
- tracing_magfloat
Magnitude by which the learning trace is increased for each spike.
- tracing_tauint
Decay time constant for the learning trace, in timesteps (not seconds).
- weights(n_axons,) list of (n_populations, n_compartments) ndarray
The synapse weights. Organized as a list of arrays so each axon can have a different number of target compartments.
-
axon_compartment_base
(axon_idx)[source]¶ Offset for compartment indices for a particular axon.
A return value of
None
indicates the axon is unused.
-
axon_weights_indices
(axon_idx, atom=0)[source]¶ The weights and indices for a particular axon (and atom, if applicable).
-
max_ind
()[source]¶ The maximum compartment index in weight memory.
Does not include
axon_compartment_base
.
-
class
nengo_loihi.probe.
LoihiProbe
(target=None, key=None, slice=None, weights=None, synapse=None, reindexing=None)[source]¶ Record data from one or more LoihiBlock target states.
To get the final output of the probe:
The LoihiBlock state given by
key
will be collected from each target.The slices will be applied to the outputs.
The weights will be applied to the sliced outputs.
If the weights change the output shape then weighted outputs are summed. If no weights or scalar weights, the weighted outputs are concatenated.
Reindexing is applied to order the outputs correctly (in the case of multiple targets with no transformitive weights).
Outputs are filtered using
synapse
.
The difference between summing and concatenating based on transformitive weights is to facilitate splitting blocks: Splitting a block with weights splits the weight matrix across the inputs but not across outputs, thus it is natural to sum outputs. Splitting a block with scalar weights results in new blocks with scalar weights, thus to get the same size of output as before the individual outputs are concatenated.
- Parameters
- targetLoihiBlock or list of LoihiBlock
The block to record values from. Use
slice
to record from a subset of compartments.- keystring (‘current’, ‘voltage’, ‘spiked’)
The compartment attribute to probe.
- slicelist of <slice or list>
Select a subset of the compartments in each block to record from.
- synapsenengo.synapses.Synapse
A synapse to use for filtering the probe.
- weightsnp.ndarray
A linear transformation to apply to the outputs.
- reindexingnp.ndarray
A list of indices used to reorder the outputs.
-
weight_outputs
(outputs)[source]¶ Apply weights and reindexing to the target outputs.
We assume that probe slices have already been applied, since these are typically performed when collecting the target outputs
- Parameters
- outputslist of lists or arrays (n_blocks, n_timesteps (optional), n_outputs)
Outputs of the target blocks. The
timesteps
dimension is optional, and defaults to 1 if not provided.n_outputs
can be different across blocks.
- Returns
- result(n_timesteps, n_outputs)
Discretization¶
-
nengo_loihi.builder.discretize.
discretize_model
(model)[source]¶ Discretize a
Model
in-place.Turns a floating-point
Model
into a discrete (integer) model appropriate for Loihi.- Parameters
- model
Model
The model to discretize.
- model
-
nengo_loihi.builder.discretize.
discretize_block
(block)[source]¶ Discretize a
LoihiBlock
in-place.Turns a floating-point
LoihiBlock
into a discrete (integer) block appropriate for Loihi.- Parameters
- block
LoihiBlock
The block to discretize.
- block
-
nengo_loihi.builder.discretize.
discretize_compartment
(comp, w_max)[source]¶ Discretize a
Compartment
in-place.Turns a floating-point
Compartment
into a discrete (integer) block appropriate for Loihi.- Parameters
- comp
Compartment
The compartment to discretize.
- w_maxfloat
The largest connection weight in the
LoihiBlock
containingcomp
. Used to set several scaling factors.
- comp
-
nengo_loihi.builder.discretize.
discretize_synapse
(synapse, w_max, w_scale, w_exp)[source]¶ Discretize a
Synapse
in-place.Turns a floating-point
Synapse
into a discrete (integer) block appropriate for Loihi.- Parameters
- synapse
Synapse
The synapse to discretize.
- w_maxfloat
The largest connection weight in the
LoihiBlock
containingsynapse
. Used to scale weights appropriately.- w_scalefloat
Connection weight scaling factor. Usually computed by
discretize_compartment
.- w_expfloat
Exponent on the connection weight scaling factor. Usually computed by
discretize_compartment
.
- synapse
-
nengo_loihi.builder.discretize.
discretize_weights
(synapse_cfg, w, dtype=<class 'numpy.int32'>, lossy_shift=True, check_result=True)[source]¶ Takes weights and returns their quantized values with weight_exp.
The actual weight to be put on the chip is this returned value divided by the
scale
attribute.- Parameters
- wfloat ndarray
Weights to be discretized, in the range -255 to 255.
- dtypenp.dtype, optional (Default: np.int32)
Data type for discretized weights.
- lossy_shiftbool, optional (Default: True)
Whether to mimic the two-part weight shift that currently happens on the chip, which can lose information for small weight_exp.
- check_resultsbool, optional (Default: True)
Whether to check that the discretized weights fall in the valid range for weights on the chip (-256 to 255).
Validation¶
Emulator¶
Software emulator for Loihi chip behaviour. |
|
Provide information about all the LoihiBlocks in the model. |
|
Base class for aspects of the emulator state. |
|
State representing the Compartments of all blocks. |
|
State representing the noise parameters for all compartments. |
|
State representing all synapses. |
|
State representing all (output) Axons. |
|
State representing all probes. |
-
class
nengo_loihi.emulator.
EmulatorInterface
(model, seed=None)[source]¶ Software emulator for Loihi chip behaviour.
- Parameters
- modelModel
Model specification that will be simulated.
- seedint, optional (Default: None)
A seed for all stochastic operations done in this simulator.
-
class
nengo_loihi.emulator.interface.
BlockInfo
(blocks)[source]¶ Provide information about all the LoihiBlocks in the model.
- Attributes
- dtypedtype
Datatype of the blocks. Either
np.float32
if the blocks are not discretized ornp.int32
if they are. All blocks are the same.- blockslist of LoihiBlock
List of all the blocks in the model.
- n_compartmentsint
Total number of compartments across all blocks.
- slicesdict of {LoihiBlock: slice}
Maps each block to a slice for that block’s compartments with respect to all compartments. Used to slice into any array storing data across all compartments.
-
class
nengo_loihi.emulator.interface.
IterableState
(block_info, block_key, strict=True)[source]¶ Base class for aspects of the emulator state.
This class takes the name of a LoihiBlock attribute as the
block_key
and maps these objects to their parent blocks and slices.- Attributes
- dtypedtype
Datatype of the state elements (given by the BlockInfo datatype).
- block_mapdict of {item: block}
Maps an item (determined by
block_key
) to the block it belongs to.- n_compartmentsint
The total number of neuron compartments (given by BlockInfo).
- slicesdict of {item: slice}
Maps an item to the
block_info.slice
for the block it belongs to.- strictbool (Default: True)
Whether “undesired” chip effects (ex. overflow) raise errors (
True
) or whether they only raise warnings (False
).
-
class
nengo_loihi.emulator.interface.
CompartmentState
(block_info, strict=True)[source]¶ State representing the Compartments of all blocks.
-
class
nengo_loihi.emulator.interface.
NoiseState
(block_info)[source]¶ State representing the noise parameters for all compartments.
-
class
nengo_loihi.emulator.interface.
SynapseState
(block_info, pes_error_scale=1.0, strict=True)[source]¶ State representing all synapses.
- Attributes
- pes_error_scalefloat
Scaling for the errors of PES learning rules.
- pes_errors{Synapse: ndarray(n_neurons / 2)}
Maps synapse to PES learning rule errors for those synapses.
- spikes_in{Synapse: list}
Maps synapse to a queue of input spikes targeting those synapses.
- traces{Synapse: ndarray(Synapse.n_axons)}
Maps synapse to trace values for each of their axons.
- trace_spikes{Synapse: set}
Maps synapse to a queue of input spikes waiting to be added to those synapse traces.
-
class
nengo_loihi.emulator.interface.
AxonState
(block_info)[source]¶ State representing all (output) Axons.
-
class
nengo_loihi.emulator.interface.
ProbeState
(block_info, probes, dt)[source]¶ State representing all probes.
- Attributes
- dtfloat
Time constant of the Emulator.
- filters{nengo_loihi.Probe: function}
Maps Probes to the filtering function for that probe.
- filter_pos{nengo_loihi.Probe: int}
Maps Probes to the position of their filter in the data.
- block_probes{nengo_loihi.Probe: slice}
Maps Probes to the BlockInfo slice for the block they are probing.
- input_probes{nengo_loihi.Probe: SpikeInput}
Maps Probes to the SpikeInput that they are probing.
Hardware¶
Place a Model onto a Loihi board and run it. |
-
class
nengo_loihi.hardware.
HardwareInterface
(model, use_snips=True, seed=None, snip_max_spikes_per_step=50, n_chips=2, allocator=None)[source]¶ Place a Model onto a Loihi board and run it.
- Parameters
- modelModel
Model specification that will be placed on the Loihi board.
- use_snipsboolean, optional (Default: True)
Whether to use snips (e.g., for
precompute=False
).- seedint, optional (Default: None)
A seed for stochastic operations.
- snip_max_spikes_per_stepint
The maximum number of spikes that can be sent to each chip in one timestep if
.use_snips
is True.- n_chipsint, optional (Default: 2)
The number of chips on the board.
- allocatorAllocator, optional (Default:
GreedyInterchip()
) Callable object that allocates the board’s devices to given models. Defaults to one block and one input per core on a single chip.
Allocators¶
Assigns each block to distinct cores on as few chips as possible. |
|
Assigns each block to distinct cores on as many chips as possible. |
DVS¶
|
Process for DVS input to Loihi chip from a pre-recorded file. |
A group of events from a Dynamic Vision Sensor (DVS) spiking camera. |
-
class
nengo_loihi.dvs.
DVSFileChipProcess
(file_path, file_fmt=None, t_start=0, rel_time=None, pool=(1, 1), channels_last=True, dvs_height=180, dvs_width=240, **kwargs)[source]¶ Process for DVS input to Loihi chip from a pre-recorded file.
- Parameters
- file_pathstring
The path of the file to read from. Can be a
.aedat
or.events
file.- file_fmt“aedat” or “events” or None, optional
The format of the file. If
None
(default), this will be detected from the file extension.- t_startfloat, optional
Offset for the time in the file to start at, in seconds.
- rel_timebool, optional
Whether to make all times relative to the first event, or not. Defaults to True for
.aedat
files and False otherwise.- pool(int, int), optional
Number of pixels to pool over in the vertical and horizontal directions, respectively.
- channels_lastbool, optional
Whether to make the channels (i.e. the polarity) the least-significant index (True) or the most-significant index (False).
- dvs_heightint, optional
The actual height (in pixels) of the DVS sensor. Only change this if your sensor has a non-standard height. If you wish to make the output of this node smaller, use the
pool
argument instead.- dvs_widthint, optional
The actual width (in pixels) of the DVS sensor. Only change this if your sensor has a non-standard width. If you wish to make the output of this node smaller, use the
pool
argument instead.- **kwargs
Extra arguments to pass to the
nengo.Process
constructor.
Examples
This example shows how to create the process, use it in a
Node
, and connect it to neurons on the Loihi chip. The DVS events loaded from the file will be transferred immediately to the Loihi chip; none of the simulation is on the host.with nengo.Network() as net: dvs_process = nengo_loihi.dvs.DVSFileChipProcess("my-dvs-events.aedat") u = nengo.Node(dvs_process) ens = nengo.Ensemble(dvs_process.size, dimensions=1) nengo.Connection(u, ens.neurons)
-
make_step
(shape_in, shape_out, dt, rng, state)[source]¶ Make the step function to display the DVS events as image frames.
This step function is only called when using this process in a
nengo.Simulator
. When using it in anengo_loihi.Simulator
, the events are transferred directly to the Loihi board.
-
class
nengo_loihi.dvs.
DVSEvents
[source]¶ A group of events from a Dynamic Vision Sensor (DVS) spiking camera.
- Attributes
- eventsstructured
numpy.ndarray
A structured array with the following fields:
“y”: The vertical coordinate of the event.
“x”: The horizontal coordinate of the event.
“p”: The polarity of the event (
0
for off,1
for on).“v”: The event trigger (
0
for DVS events,1
for external events).“t”: The event timestamp in microseconds.
n_events
intThe number of events (equals
len(self.events)
).
- eventsstructured
-
property
n_events
¶ The number of events (equals
len(self.events)
).
-
static
from_file
(file_path, **kwargs)[source]¶ Create a new
DVSEvents
object with events from a file.- Parameters
- file_pathstr
The path to the events file.
- **kwargs
Additional keyword arguments to pass to
DVSEvents.read_file
.
-
init_events
(event_data=None, n_events=None)[source]¶ Initialize
events
array.Only required if configuring events manually (i.e. not reading from a file).
- Parameters
- event_datalist of tuples, optional
Each tuple is of the form
(y, x, p, v, t)
. SeeDVSEvents
for the definitions of these fields.- n_eventsint, optional
The number of events in the new (empty) events array, in the case that
event_data
is not provided.
-
read_file
(file_path, file_fmt=None, rel_time=None)[source]¶ Read events from a file.
- Parameters
- file_pathstr
The path to the events file.
- file_fmt“aedat” or “events” or None, optional
The file format of the events file. If
None
, will be detected based on the file extension.- rel_timebool, optional
Whether timestamps should be relative to the first event, or absolute.