张量流除以 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)