【pytorch函数笔记】torch.nn.Sigmod

转载：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位小数。