Spiking Neural Network

class snnpytorch.network.spiking_neural_network.SNN(num_input_neurons=10, num_output_neurons=100, conn_prob=0.5)

Bases: torch.nn.modules.module.Module

Class of a spiking neural network

Parameters

  • num_input_neurons – Number of neurons in input layer

  • num_output_neurons – Number of neurons in output layer

  • conn_prob – Connection probability from input to output layer

forward(input_spikes: torch.Tensor) → torch.Tensor

Forward pass for this spiking neuron network

Parameters

input_spikes – Network input data (spike raster)

Returns

Network output data (spike raster)