2020年2月4日
PyTorch学习笔记6--案例2:PyTorch神经网络(MNIST CNN)
摘要： 上一节中，我们使用autograd的包来定义模型并求导。本节中,我们将使用torch.nn包来构建神经网络。 一个nn.Module包含各个层和一个forward(input)方法，该方法返回output. 上图是一个简单的前馈神经网络。它接受一个输入。然后一层接着一层地传递。最后输出计算的结果。 阅读全文
posted @ 2020-02-04 16:01 云远·笨小孩 阅读(4857) 评论(2) 推荐(0) 编辑