Currently we only have a load function. But after training the network it would be great if we can save the generated tensor to a file. @soumith I think we can reuse a lot of code from this repo https://github.com/MattVitelli/GRUV