趋近智
requires_grad
)backward()
).grad
)torch.nn
搭建模型torch.nn.Module
基类torch.nn
损失)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
这部分内容有帮助吗?
torch.nn
- PyTorch Documentation, PyTorch Authors, 2024 - PyTorch神经网络模块的官方文档,涵盖模型定义、层以及诸如交叉熵损失和均方误差损失等多种损失函数。torch.optim
- PyTorch Documentation, PyTorch Authors, 2025 (PyTorch Foundation) - PyTorch优化算法的官方文档,包含随机梯度下降(SGD)和Adam等优化器,并解释了它们的用法和参数。