undefined min read
PyTorch Internals 06 - When and How to Use a Custom Autograd Function
Custom autograd functions are a practical place to define forward-backward contracts before dropping to lower-level extensions
Custom autograd functions are a practical place to define forward-backward contracts before dropping to lower-level extensions