I wanted to write a custom layer in Tensorflow in which I would need to save previous weights of the layer (so for each of the last n epoch). Where in the custom layer would