趋近智
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
模块文档,提供了神经网络的构建模块,包括示例中使用的nn.Conv2d
、nn.MaxPool2d
、nn.ReLU
和nn.Linear
层。torch.nn.Module
和本节讨论的层构建神经网络,包括一个简单的CNN,从而加强实际操作。