本博客参考 http://studyai.com/pytorch-1.4/beginner/blitz/neural_networks_tutorial.html + 《PyTorch入门与实践》
torch.nn最核心的类是nn.Module,一个 nn.Module
包含若干 layers, 和一个方法 forward(input)
, 该方法返回 output
。
标签:1.4,nn,beginner,torch,blitz,Module,神经网络
来源: https://www.cnblogs.com/xixixing/p/12626987.html