PyTorch中 torch.nn与torch.nn.functional的区别

原文链接

torch.nn

pytorch中文文档链接:torch.nn

在__init__()函数里定义,定义的是一个类:

torch.nn.functional

pytorch中文文档链接:torch.nn.functional

在__forward()__函数里定义,定义的是一个函数:

两者的区别:

torch.nn中是一个定义的类,以class xx来定义的,可以提取变化的学习参数。

torch.nn.functional是一个函数,由def function( )定义,是一个固定的运算公式。

深度学习中会有很多权重是在不断更新的,所以需要采用类的方式,以确保能在参数发生变化时仍能使用我们之前定好的运算步骤。因此如果模型有可学习的参数,应该使用nn.Module,否则两个没有区别。但是简单的计算不需要新建一个类来做,所以使用nn.functional定义函数就可以。

即:层内有variable的情况用nn定义,否则用nn.functional定义。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容