转载:https://blog.csdn.net/qq_43115981/article/details/115357394
一、sigmoid
介绍
sigmoid是激活函数的一种,它会将样本值映射到0到1之间。
sigmoid的公式如下:
二、sigmoid
的应用
import torch
import torch.nn as nn
import math
a = torch.arange(4).reshape(2,2).float()
a
b =nn.Sigmoid()
b(a)
1/(1+pow(math.e, -1*0))
1/(1+pow(math.e, -1*1))
1/(1+pow(math.e, -1*2))
1/(1+pow(math.e, -1*3))
结果:
torch.nn.Sigmod结果与公式一致,函数保留4位小数。