python - Tensorflow.py protected 划分

Question

我正在尝试使用 Tensorflow.where 实现一种 protected 除法，但不知何故它似乎跳过了 where 语句上设置的条件。

主要思想是，在除 x/y 时，如果 y == 0. 则除法的结果为 x而不是抛出错误。

我的代码如下:

def Pdivide(x,y):
    result = tf.where(y == 0., x, x/y) 
    return result

但不知何故，这个条件被跳过了:

>>> a = tf.Variable([1.7, 0.2, 0., 1.1, 0.9, 0.3, 23., -1.])
>>> b = tf.Variable([0., 0., 0., 1., 1., 0., 1., 1.])

>>>Pdivide(a,b)

>>>(inf, inf, nan, 1.1, 0.9, inf, 23, -1)

预期输出:

>>>(1.7, 0.2, 0., 1.1, 0.9, 0.3, 23, -1)

PS:使用eager执行。

Answer 1

好吧，答案显然很简单。

由于某种原因，张量元素无法与简单的 == 进行比较，但使用 tf.equal(y, 0.) 可以解决问题并产生正确的输出。