python - tensorflow 除以 0/0= :0

Question

我要师归还0.为 0./0.而不是 NaN或 tensorflow 应用程序中的错误。

我知道如何在 numpy 中做到这一点 [1] , [2] ，但我是 tensorflow 的新手。

如何做到这一点？

Answer 1

这个问题是2年前问的，当时不确定这个API是否支持，但是现在Tensorflow 2.X真的支持了:

#Computes a safe divide which returns 0 if the y is zero.
tf.math.divide_no_nan(
    x, y, name=None
)

参数:

x:张量。必须是以下类型之一:float32、float64。

y:dtype 与 x 兼容的张量。

name:操作的名称(可选)。

返回:

x 除以 y 的元素值。

需要注意参数类型，应该只有tf.float32或tf.float64，如果是tf.int*，tf2.x会报错。以下是我在 colab 中正确运行的测试代码:

import tensorflow as tf
myShape=(30,30)
a = tf.constant(2, shape=myShape, dtype=tf.float32)
z = tf.constant(0, shape=myShape, dtype=tf.float32 )
cz2 = tf.math.divide_no_nan(a, z)
print(cz2)