API symbolsautograd
Copy
tensorplay.autograd.function.once_differentiable
- tensorplay.autograd.function.once_differentiable(fn)[source]
Decorator to make a custom autograd Function’s backward run once, with gradients detached and grad-mode disabled inside.
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?