PyTorch Forums
When should you save_for_backward vs. storing in ctx
tom
(Thomas V)
October 2, 2018, 7:45am
8
Hu? The ctx would go out of scope after the backward, too, would it not?
Best regards
Thomas
How to save a list of integers for backward when using CPP custom layer?
Memory management for variables stored during forward pass
show post in topic