torch.sigmoid、torch.nn.Sigmoid和torch.nn.functional.sigmoid的区别

10 篇文章 0 订阅
3 篇文章 0 订阅

  review代码的时候发现我使用的是torch.sigmoid,pycharm标黄cannot find reference 'sigmoid' in '__init__.py'于是产生了这样的疑问,这三者的区别是啥。

0x00. 官网解释

  • torch.sigmoid
    torch.sigmoid
      我们可以看到,这是一个方法,拥有Parametrs和Returns。
  • torch.nn.Sigmoid

官方解释

官方解释
  可以看到官网文档在左上角标注着显眼的CLASS,同时根据Examples我们可以得出结论,torch.nn.Sigmoid在我们的神经网络中使用时,我们应该将其看作是网络的一层而不是简单的函数使用

  • torch.nn.functional.sigmoid
    官方解释
      事实上,torch.nn.functional从这个包名就能看出来,这个包里的都是函数。同样的,按照官网的文档的内容,我们也可以判断出torch.nn.funtional.sigmoid是一个方法,可以直接在我们的神经网络的forward中使用,并不需要在init的时候初始化。也就是说torch.nn.functional.sigmoid和torch.sigmoid没有什么区别,同理,本文对于其他的激活函数一样适用。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值