Skip to main content
Modern PyTorch Guide home page
Search...
⌘K
Official Docs
GitHub
GitHub
Search...
Navigation
Neural Network Basics
nn.Module Basics
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
nn.Module Basics
Neural Network Basics
nn.Module Basics
The building block of all neural networks in PyTorch
nn.Module Basics
Creating custom modules,
init
, forward(), and module composition.
Forward mode
Parameters & Buffers
⌘I
