I have the same question (I need to store info for backward).
I am following this example: http://pytorch.org/tutorials/beginner/examples_autograd/two_layer_net_custom_function.html 166,
Pytorch 0.2 and get the same RuntimeError
Unfortunately,  I don’t see where this requirement (store only input or output) comes from and how do I get around it (storing as a member in Function means you can use this function only once).
Any ideas how to store intermediate things?