You could implement a buffer, but I don’t think that manipulating it in the forward would reflect the changes in the original model.module, since it would be unclear which value and shape (in the second iteration) to store.
Could you explain your use case a bit? Maybe there is another way to achieve, what you need.