Replace the automatic autograd with your own torch.autograd.Function: Revision history

From Master of Neuroscience Wiki

Diff selection: Mark the radio buttons of the revisions to compare and hit enter or the button at the bottom.
Legend: (cur) = difference with latest revision, (prev) = difference with preceding revision, m = minor edit.

21 October 2025

  • curprev 14:5214:52, 21 October 2025 Davrot talk contribs 10,744 bytes +10,744 Created page with "Questions to [mailto:davrot@uni-bremen.de David Rotermund] == [https://pytorch.org/docs/stable/notes/extending.html#extending-autograd Extending torch.autograd] == === [https://pytorch.org/docs/stable/notes/extending.html#when-to-use When to use] === <blockquote>In general, implement a custom function if you want to perform computations in your model that are not differentiable or rely on non-PyTorch libraries (e.g., NumPy), but still wish for your operation to chain w..." Tag: Visual edit