PyTorch学习系列(三)——构建神经网络

本文将介绍如何用PyTorch构建自己的神经网络模型。

torch.nn.Module类是PyTorch中用于表示多层网络的。构建自己的网络就是定义torch.nn.Module的一个子类，这时需要重写初始化函数__init__()和前向过程forward()。__init__()函数中需要调用父类的初始化函数；forward()函数用于构建网络从输入到输出的过程。

示例
假设我们想构建两层的全连接神经网络，输入大小为100，第一层网络的神经元数量为200， 第二层网络的神经元数量为10：

import torch.nn as nn
import torch.nn.functional as F
class MyNet(nn.Module):
    def __init__(self):
        super(MyNet, self).__init__()