norse.torch.utils

norse.torch.utils#

Utilities for spiking neural networks based on PyTorch.

Packages and subpackages may depend on Matplotlib and Tensorboard.

Plotting#

plot_heatmap_2d

Plots a heatmat of two-dimensional data

plot_heatmap_3d

Plots heatmaps for some activity in several layers.

plot_histogram_2d

Plots a histogram of 1-dimensional data.

plot_izhikevich

Computes and plots a 2D visualisation of the behavior of an Izhikevich neuron model.

plot_neuron_states

Plots state variables in a line plot based on a list of states over time.

plot_scatter_3d

Plots spike activity in time.

plot_spikes_2d

Plots a 2D diagram of spikes.

Tensorboard#

hook_spike_activity_mean

Generates a hook that can be applied to stateful torch Modules.

hook_spike_activity_sum

Generates a hook that can be applied to stateful torch Modules.

hook_spike_histogram_mean

Generates a hook that can be applied to stateful torch Modules.

hook_spike_histogram_sum

Generates a hook that can be applied to stateful torch Modules.

hook_spike_image

Generates a hook that can be applied to stateful torch Modules.