当前位置:   article > 正文

【pytorch函数笔记】torch.nn.Sigmod_nn.sigmoid

nn.sigmoid

转载:https://blog.csdn.net/qq_43115981/article/details/115357394

一、sigmoid介绍

sigmoid是激活函数的一种,它会将样本值映射到0到1之间。
sigmoid的公式如下:

\frac{1}{1+e^{-x}}

二、sigmoid的应用

  1. import torch
  2. import torch.nn as nn
  3. import math
  4. a = torch.arange(4).reshape(2,2).float()
  5. a
  6. b =nn.Sigmoid()
  7. b(a)
  8. 1/(1+pow(math.e, -1*0))
  9. 1/(1+pow(math.e, -1*1))
  10. 1/(1+pow(math.e, -1*2))
  11. 1/(1+pow(math.e, -1*3))

结果:

 torch.nn.Sigmod结果与公式一致,函数保留4位小数

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/205953
推荐阅读
相关标签
  

闽ICP备14008679号