Python — From Basics to Advanced · PyTorch — Tensors, Autograd, and a Simple MLP
Tensor — NumPy + GPU + autograd
PyTorch — Tensors, Autograd, and a Simple MLP
Introduction
torch.Tensor looks like np.ndarray, but has three superpowers: it runs on GPU (one .to(device) call and you're set), supports autograd (gradients automatically), and integrates with nn.Module.