To summarise my previous question:


How do I compute backprop gradients for a given batch but, not apply those gradients to the model automatically?


How can I do optimiser.step() for the gradients I manually apply to the model?