Pytorch 之torch.nn初探 第1关:torch.nn.Module

神经网络可以使用torch.nn包构建。它提供了几乎所有与神经网络相关的功能,例如:

  • 线性图层 nn.Linearnn.Bilinear
  • 卷积层 nn.Conv1dnn.Conv2dnn.Conv3dnn.ConvTranspose2d
  • 非线性 nn.Sigmoidnn.Tanhnn.ReLUnn.LeakyReLU
  • 池化层 nn.MaxPool1dnn.AveragePool2d
  • Recurrent网络 nn.LSTMnn.GRU
  • 标准化 nn.BatchNorm2d
  • Dropout nn.Dropoutnn.Dropout2d
  • Embedding - nn.Embedding
  • 损失函数 nn.MSELossnn.CrossEntropyLossnn.NLLLoss

这些类的实例将具有一个内置的__call__函数,可通过图层运行输入。

任务描述

本关任务:

本关要求利用nn.Linear()

你可能感兴趣的:(educoder人工智能答案)