其他分享
首页 > 其他分享> > Pytorch中nn.Moudle模块和nn.functional模块库的不同

Pytorch中nn.Moudle模块和nn.functional模块库的不同

作者:互联网

nn.Moudle模块库中存放着很多和神经网络相关的工具,而nn.functional里也一样提供了很多网络层相关的函数功能。
最重要的不同点在于,nn.functional里定义的网络层不可自动学习参数,还需要使用nn.Parameter封装。nn.functional设计的初衷是为了一些不需要学习参数的层,比如激活层,BN层这些,这样就不用在nn.Moudle里定义了。

标签:定义,nn,Moudle,网络层,functional,模块
来源: https://blog.csdn.net/weixin_39977764/article/details/117221255