Each call to .apply will have a different context. So you can save everything you need in it without risk.
Note that if you need to save input or outputs, you should use the ctx.save_for_backward.
Each call to .apply will have a different context. So you can save everything you need in it without risk.
Note that if you need to save input or outputs, you should use the ctx.save_for_backward.