个性化阅读
专注于IT技术分析

标签:torch.nn

Python

PyTorch中的torch.nn模块使用详解

半瓶木阅读(10736)评论(0)赞(1)

PyTorch提供了torch.nn模块来帮助我们创建和训练神经网络。我们将首先在MNIST数据集上训练基本神经网络, 而无需使用这些模型的任何功能。我们将仅使用基本的PyTorch张量功能, 然后一次从torch.nn中增量添加一个功能。...