Thank you for the quick response.
This is really helpful. However, my gradchecks passes what could be the reason for that.
Also, to clarify as I am new to python, saving with ctx means similar to self?
Example:
ctx.intermediate_result = torch.Tensor(2, 3)
Thanks!