赞
踩
神经网络可以使用torch.nn
包构建。它提供了几乎所有与神经网络相关的功能,例如:
nn.Linear,nn.Bilinear
nn.Conv1d,nn.Conv2d,nn.Conv3d,nn.ConvTranspose2d
nn.Sigmoid,nn.Tanh,nn.ReLU,nn.LeakyReLU
nn.MaxPool1d,nn.AveragePool2d
nn.LSTM,nn.GRU
nn.BatchNorm2d
Dropout nn.Dropout,nn.Dropout2d
Embedding - nn.Embedding
nn.MSELoss,nn.CrossEntropyLoss,nn.NLLLoss
这些类的实例将具有一个内置的__call__函数,可通过图层运行输入。
本关任务:
本关要求利用nn.Linear()声明一个线性模型 l,并构建一个变量 net 由三个l序列构成。
torch.nn.Module是所有神经网络模块的基类,用户自定义的神经网络模型同样继承自这个类。它定义了训练神经网络需要的所有基础方法,并且是可以序列化的抽象类。
tor
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。