一、神经网络工具箱nn
torch.nn
是专门为深度学习设计的模块。torch.nn
的核心数据结构是Module, 他是一个抽象的概念,既可以表示神经网络的某个层(layer),也可以表示一个包含很多层的神经网络。
在实际使用中,最常见的做法是继承nn,Module,撰写自己及的网络/层。
下面来看看如何使用nn.Module实现自己的全连接层。
全连接层,又名仿射层,输出 y y 和输入
满足 y=
torch.nn
是专门为深度学习设计的模块。torch.nn
的核心数据结构是Module, 他是一个抽象的概念,既可以表示神经网络的某个层(layer),也可以表示一个包含很多层的神经网络。
在实际使用中,最常见的做法是继承nn,Module,撰写自己及的网络/层。
下面来看看如何使用nn.Module实现自己的全连接层。
全连接层,又名仿射层,输出 y y 和输入
满足 y=