Tensorflow替换张量中接近零的值

Question

我想避免代码中包含许多除法的数字不稳定性。如何用预定义的数值稳定器替换张量中的小值？

让稳定器为EPS=1e-9。 T是一个张量，其中包含许多接近零的小值。我想用[-EPS,EPS]替换区间EPS中的所有值。我该怎么办？

Answer 1

您可以执行以下操作：

 my_tensor_stable = tf.where(tf.abs(my_tensor) < EPS,
                             EPS * tf.ones_like(my_tensor),
                             my_tensor)