本博客参考 http://studyai.com/pytorch-1.4/beginner/blitz/neural_networks_tutorial.html + 《PyTorch入门与实践》
torch.nn最核心的类是nn.Module,一个 nn.Module
包含若干 layers, 和一个方法 forward(input)
, 该方法返回 output
。
2024-04-01 18:08:52
本博客参考 http://studyai.com/pytorch-1.4/beginner/blitz/neural_networks_tutorial.html + 《PyTorch入门与实践》
torch.nn最核心的类是nn.Module,一个 nn.Module
包含若干 layers, 和一个方法 forward(input)
, 该方法返回 output
。