使用pytorch实现sigmoid函数
时间: 2023-12-09 13:04:15 浏览: 131
基于Python的sigmoid函数FPGA实现.zip
5星 · 资源好评率100%
可以使用torch.sigmoid()函数来实现sigmoid函数。例如:
import torch
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.sigmoid(x)
print(y)
输出结果为:tensor([0.7311, 0.8808, 0.9526])
阅读全文