PyTorch Visual

Interactive, step-by-step visualizations for every PyTorch operation. Tweak inputs, watch animations, and build intuition for how tensors transform.

Tensor Basics
7 operations
torch.tensor()torch.zeros()torch.ones()torch.rand()torch.arange()Tensor.shape / .size()unsqueeze() / squeeze()
Math Operations
4 operations
torch.add()torch.mul()torch.sum()torch.mean()
Linear Algebra
1 operations
torch.matmul()
Reshaping
3 operations
torch.reshape()Tensor.view()Tensor.permute()
Indexing
1 operations
Indexing & Slicing
Activations
2 operations
nn.ReLU()Softmax
Autograd
1 operations
Autograd
NN Layers
2 operations
nn.Linear()F.conv2d()
Loss Functions
2 operations
CrossEntropyLossMSELoss
Optimization
1 operations
SGD / Adam Optimizer