其他分享
首页 > 其他分享> > 4、神经网络的模块化接口torch.nn

4、神经网络的模块化接口torch.nn

作者:互联网

本博客参考 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