memory_layer function

A torch::nn_module() Representing a Memory Layer