神经网络工具箱
autograd虽然可以实现深度学习模型,但是抽象程度低,编写代码量大。这种情况下,torch.nn应运而生。
torch.nn的核心数据结构是Module,这是一个抽象概念,既可以表示神经网络中的某个层,也可以表示含多个层的神经网络。
最常见的做法是,继承nn.Module,撰写自己的网络/层。下面利用nn.Module实现自己的全连接层。
输入y与输出x满足
y=Wx+b y = W x + b
神经网络工具箱
autograd虽然可以实现深度学习模型,但是抽象程度低,编写代码量大。这种情况下,torch.nn应运而生。
torch.nn的核心数据结构是Module,这是一个抽象概念,既可以表示神经网络中的某个层,也可以表示含多个层的神经网络。
最常见的做法是,继承nn.Module,撰写自己的网络/层。下面利用nn.Module实现自己的全连接层。
输入y与输出x满足