当前位置:   article > 正文

Pytorch 之torch.nn初探--第1关:torch.nn.Module_头歌pytorch之torch初探答案

头歌pytorch之torch初探答案

神经网络可以使用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
  • Recurrent网络 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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/716957
推荐阅读
相关标签
  

闽ICP备14008679号