关于pytorch:PyTorch中-torchnn与torchnnfunctional的区别

原文链接

torch.nn

pytorch中文文档链接：torch.nn

在__init__()函数里定义，定义的是一个类：

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

在__forward()__函数里定义，定义的是一个函数：

torch.nn中是一个定义的类，以class xx来定义的，能够提取变动的学习参数。

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

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

即：层内有variable的状况用nn定义，否则用nn.functional定义。

学习更多编程常识，请关注我的公众号：

代码的路