torch nn functional,pytorch与torchvision

  torch nn functional,pytorch与torchvision

  torch.nn

  Pytorch中文文档链接:torch.nn

  在__init__()函数中定义,该函数定义了一个类:

  火炬. nn .功能

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

  在__forward()__函数中定义,它定义了一个函数:

  两者的区别

  Torch.nn是一个定义类,由xx类定义,可以提取变化的学习参数。

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

  在深度学习中,会有很多不断更新的权重,所以我们需要采用类的方式,保证在参数发生变化时,仍然可以使用之前设定的操作步骤。因此,如果模型具有可学习的参数,nn。模块,否则两者没有区别。但是简单的计算不需要创建新的类,可以用nn.functional来定义函数。

  也就是说,如果层中有变量,则由nn定义,否则由nn.functional定义

  了解更多编程知识,请关注我的微信官方账号:

  编码道路

  原创作品来自博客作者代码之路,

郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: