pytorch

8963d771 - Clarify the saving of intermediates in the "extending torch.func" docs (#98020) (#98081)

Clarify the saving of intermediates in the "extending torch.func" docs (#98020) (#98081) Fixes https://github.com/pytorch/pytorch/issues/97260 We got some feedback that the page reads like "in order to save an input for backward, you must return it as an output of the autograd.Function.forward". Doing so actually raises an error (on master and as of 2.1), but results in an ambiguous situation on 2.0.0. To avoid more users running into this, we clarify the documentation so it doesn't read like the above and clearly mentions that you can save things from the inputs or outputs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/98020 Approved by: https://github.com/soulitzer, https://github.com/kshitij12345