requires_grad
)backward()
).grad
)torch.nn
torch.nn.Module
Base Classtorch.nn
losses)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
Was this section helpful?
torch.nn
- PyTorch Documentation, PyTorch Authors, 2024 - Official documentation for PyTorch's neural network module, covering model definitions, layers, and various loss functions like CrossEntropyLoss and MSELoss.torch.optim
- PyTorch Documentation, PyTorch Authors, 2025 (PyTorch Foundation) - Official documentation for PyTorch's optimization algorithms, including Stochastic Gradient Descent (SGD) and Adam, explaining their usage and parameters.