Sigmoid函数指数运算溢出问题,tensorflow版本代码
对于tensor张量类别的数据,如何在tensorflow环境中进行运算,避免运算溢出。在本站我看了很多人写的代码,有的方案能对单个数计算,有的方案能在torch环境中计算,但是都不符合我在tensorflow环境中使用的要求,于是尝试写了一个比较简单的运算方法。
方法原理:
这个公式是目前网上比较普遍的方法。
tensorflow中代码实现
def sigmoid(x):
y=tf.where(x >= 0,(1 / (1 + tf.exp(-x))),(tf.exp(x) / (1 + tf.exp(x))))
return y
经实验验证,该方法是能够实现原理要求的。