当前位置:   article > 正文

【pytorch函数笔记(二)】torch.nn.Sigmoid()

torch.nn.sigmoid
import torch.nn as nn
torch.nn.Sigmoid()
  • 1
  • 2

一、sigmoid介绍
  sigmoid是激活函数的一种,它会将样本值映射到0到1之间。
  sigmoid的公式如下:
1 1 + e − x \frac{1}{1+e^{-x}} 1+ex1
二、sigmoid的应用
代码:

import torch.nn as nn
import torch

#取一组满足标准正态分布的随机数构成3*3的张量
t1 = torch.randn(3,3)
m = nn.Sigmoid()
t2 = m(t1)
print(t1)
print(t2)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

运行结果:

tensor([[ 0.1904,  0.5954,  1.8083],
        [ 1.5643, -0.2557,  1.8694],
        [ 0.0155, -2.0995, -1.3233]])
tensor([[0.5475, 0.6446, 0.8592],
        [0.8270, 0.4364, 0.8664],
        [0.5039, 0.1091, 0.2103]])
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/205896
推荐阅读
相关标签
  

闽ICP备14008679号