趋近智
requires_grad
)backward()
).grad
)torch.nn
搭建模型torch.nn.Module
基类torch.nn
损失)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
这部分内容有帮助吗?
torch.utils.data.DataLoader
, PyTorch Authors, 2025 (PyTorch Foundation) - 官方文档,介绍了DataLoader类、其参数以及与数据集和采样器的集成。torch.utils.data.Sampler
, PyTorch Authors, 2025 (PyTorch Foundation) - 官方文档,详细介绍了各种内置采样器并提供了创建自定义采样策略的指导。torch.nn.utils.rnn.pad_sequence
, PyTorch Authors, 2024 - 序列填充的官方文档,展示了自定义批处理函数的特定实用工具。