Pytorch之Base_calcuate(卷积层,池化层,激活函数,全连接层,顺序容器(Sequential))
torch.nn最基本的需要了解module类,该类是所有网络的基类,你的模型都得继承它类似示例中的提示,继承并复写init和forwardimporttorchfromtorchimportnnclassnew_class(nn.Module):def__init__(self):super(new_class,self).__init__()defforward(self,input):ret