Pytorch implementation of a stateful layer, to be used as base class.
Not implemented - You need to implement a forward method in child class