简介:本篇文章将详细解释PyTorch中的torch.exp()和torch.sigmoid()函数,并通过示例展示如何使用它们。
在PyTorch中,torch.exp()和torch.sigmoid()是两个常用的函数,用于计算指数和逻辑斯蒂映射。下面我们将分别介绍它们的定义、功能和示例。
torch.exp()是PyTorch中用于计算指数的函数。它接受一个张量作为输入,并返回该张量的指数值。指数函数在数学上定义为e的x次方,其中e是自然对数的底数(约等于2.71828)。
函数定义:
import torchx = torch.tensor([1.0, 2.0, 3.0])y = torch.exp(x)
功能解释:
二、torch.sigmoid()函数
import torch# 标量输入x = torch.tensor(2.0)y = torch.exp(x) # 输出:tensor(7.3890560989306495)# 向量输入x = torch.tensor([1.0, 2.0, 3.0])y = torch.exp(x)print(y) # 输出:tensor([ 2.71828175, 7.38905609, 20.08553692])
torch.sigmoid()是PyTorch中用于计算逻辑斯蒂映射的函数。逻辑斯蒂映射将输入值映射到0到1之间,常用于二分类问题的概率估计。
函数定义:
import torchx = torch.tensor([-1.0, 0.0, 1.0])y = torch.sigmoid(x)
功能解释:
import torch# 标量输入x = torch.tensor(-1.0)y = torch.sigmoid(x) # 输出:tensor(0.2689)# 向量输入x = torch.tensor([-1.0, 0.0, 1.0])y = torch.sigmoid(x)print(y) # 输出:tensor([0.2689, 0.5000, 0.7311])