【PyTorch】教程：torch.nn.Sigmoid_nn.sigmoid()

作者：2023面试高手 | 2024-03-07 13:45:45

踩

nn.sigmoid()

torch.nn.Sigmoid

原型

CLASS torch.nn.Sigmoid(*args, **kwargs)

定义

Applies the element-wise function:

$Sigmoid(x)=\sigma(x)=\frac{1}{1+\exp(-x)}$

图

在这里插入图片描述

代码

import torch
import torch.nn as nn

m = nn.Sigmoid()
input = torch.randn(4)
output = m(input)

print("input: ", input)   # input:  tensor([-0.8462,  0.7929, -0.5680, -0.5883])
print("output: ", output) # output:  tensor([0.3002, 0.6885, 0.3617, 0.3570])
1
2
3
4
5
6
7
8
9

【参考】

Sigmoid — PyTorch 2.0 documentation

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/205948