趋近智
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.Sequential
容器的官方文档,详细介绍了其功能和构建线性神经网络的方法。torch.nn.Module
基类的官方文档,解释了如何在 PyTorch 中构建自定义神经网络层和模型,是 nn.Sequential
的基础。nn.Module
和 nn.Sequential
构建神经网络,展示了它们的实际应用。