pytorch/docs
soulitzer 2f615f6313 Improve custom function docs (#60312)
Summary:
- Adds some code examples for `ctx` methods and make requirements of arguments more clear
- Type annotations for `save_for_backward`, `mark_dirty`, `mark_non_differentiable`, and `set_materialize_grads` (BC-breaking?)
- Refactor `torch.autograd.Function` doc

Pull Request resolved: https://github.com/pytorch/pytorch/pull/60312

Reviewed By: VitalyFedyunin

Differential Revision: D30314961

Pulled By: soulitzer

fbshipit-source-id: a284314b65662e26390417bd2b6b12cd85e68dc8
2021-08-18 11:31:31 -07:00
..
caffe2 Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
cpp Add copy button to code snippets in docs (#63149) 2021-08-15 06:25:32 -07:00
source Improve custom function docs (#60312) 2021-08-18 11:31:31 -07:00
.gitignore
libtorch.rst
make.bat
Makefile
README.md Add docs/README.md to make existing doc build info more discoverable (#49286) 2020-12-16 11:55:45 -08:00
requirements.txt Add copy button to code snippets in docs (#63149) 2021-08-15 06:25:32 -07:00

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.