x = torch.randn(4)
def fast_sigmoid_v2():
y1 = [1 / (1 + torch.exp(-i)) for i in x]
y2 = []
y3 = []
for i in x:
y2.append(0.5 * (i / (0.99 / torch.abs(i) + torch.abs(i)) + 1))
for i in x:
y3.append(0.5 * (i / (1 + torch.abs(i)) + 1))
print(y2)
结果
[tensor(0.2999), tensor(0.7021), tensor(0.2735), tensor(0.5029)]
Process finished with exit code 0