摘要: 神经网络可以使用 torch.nn包构建。 现在你已经对autograd有所了解,nn依赖 autograd 定义模型并对其求微分。nn.Module 包括层,和一个返回 output 的方法 - forward(input)。 例如,看看这个对数字图片进行分类的网络: convnet 这是一个简单 阅读全文
posted @ 2022-01-21 10:17 Deep_RS 阅读(254) 评论(0) 推荐(0) 编辑