StatefulLayer#

class sinabs.layers.StatefulLayer(state_names: List[str])[source]#

A base class that instantiates buffers/states which update at every time step.

Parameters

state_names (list of str) – the PyTorch buffers to initialise. These are not parameters.

forward(*args, **kwargs)[source]#

Not implemented - You need to implement a forward method in child class