PyTorch Cheat Sheet & Quick Reference
PyTorch Cheat Sheet & Quick Reference
This is a quick reference list of cheat sheets for PyTorch. See also PyTorch website
# Imports
General
ONNX
Vision
Distributed Training
# Tensors
Creation
Dimensionality
Algebra
GPU Usage
torch.cuda.is_available # check for cuda
x = x.cuda() # move x's data from
# CPU to GPU and return new object
Deep Learning
Loss Functions
Activation Functions
Optimizers
# Data Utilities
Datasets