资源说明:哎…年前就看到了这里,我的水平和这一章之间可能隔了一个太平洋(/TДT)/
torch.nn是专门为深度学习设计的模块,核心数据结构是Module,既可以表示神经网络中的某个层,也可以表示一个包含很多层的神经网络。
4.1 nn.Module
全连接层
我们可以继承nn.Module类来写一个自己的Module,这个例子中我们写的是全连接层y=Wx+b
import torch as t
from torch import nn
from torch.autograd import Variable as V
下面这段代码定义了新的类Linear,把nn.Module放在括号中表示它是Lin
本源码包内暂不包含可直接显示的源代码文件,请下载源码包。