Skip to main content
Modern PyTorch Guide home page
Search...
⌘K
Official Docs
GitHub
GitHub
Search...
Navigation
Autograd & Differentiation
Custom Autograd Functions
Foundations
Building Models
Performance
Domains
Production
Advanced
API Reference
Community
Forums
Getting Started
Overview
Study Plan 1
Study Plan 2
Tensors & Operations
Introduction to Tensors
Tensor Creation
Indexing & Slicing
Tensor Operations
Broadcasting
Views strides
Named tensors
Tensor attributes
Notes
Autograd & Differentiation
Autograd Basics
Computational Graph
Custom Autograd Functions
Gradients Explained
Requires grad
Forward mode
Neural Network Basics
nn.Module Basics
Parameters & Buffers
Forward Pass
Hooks
State dict
Init strategies
Data Loading
Datasets
DataLoaders
Transforms
Samplers
Pin memory
Webdataset
ETL Process
Overview
On this page
Custom Autograd Functions
Autograd & Differentiation
Custom Autograd Functions
Define your own forward and backward passes
Custom Autograd Functions
torch.autograd.Function for custom differentiable operations.
Computational Graph
Gradients Explained
⌘I
