models.modules

models.modules#

Custom modules and layer generators

Classes

Conv

Generator of standard 2d convolution

ConvLSTM

Convolutional LSTM

Dense

Class inherited from list type without changes

LI

Generator of the layer of LI neurons

LIF

Generator of the layer of LIF neurons

LSTM

LSTM layer generator

LayerGen

Base class for model layer generators

Norm

Batch Normalization layer generator

Pass

A placeholder layer generator that does nothing

Pool

Pooling layer generator

ReLU

ReLU layer generator

Residual

Class inherited from list type without changes

Return

Generator of layers for storing forward pass values

SiLU

SiLU layer generator

Storage

Stores the forward pass values

SumPool2d

Applies a 2D average pooling over an input signal composed of several input planes

Tanh

SiLU layer generator

Up

Upsample layer generator