现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.sigmoid 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

torch.sigmoid 是 PyTorch 中用于计算逐元素 sigmoid 函数的函数。sigmoid 函数定义为 1/(1+exp(-x))。

函数定义

torch.sigmoid(input, out=None)

使用示例

实例

import torch

# 计算 sigmoid 函数
x = torch.tensor([0.0, 1.0, -1.0])
result = torch.sigmoid(x)
print(result)

Pytorch torch 参考手册 Pytorch torch 参考手册