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.utils.data.DataLoader
, PyTorch Authors, 2025 (PyTorch Foundation) - Official documentation describing the DataLoader class, its arguments, and integration with datasets and samplers.torch.utils.data.Sampler
, PyTorch Authors, 2025 (PyTorch Foundation) - Official documentation detailing various built-in samplers and providing guidance for creating custom sampling strategies.torch.nn.utils.rnn.pad_sequence
, PyTorch Authors, 2024 - Official documentation for padding sequences, demonstrating a specific utility for custom collate functions.