Tensorflow重写tf.divide（）以除以0时返回分子

Question

是否可以覆盖tf.divide()方法，以便在除以0时返回分子而不是Inf。 ？

示例：

原始输出：

a = tf.Variable([1, 2,3,4]) b = tf.Variable([1, 0,3,4]) tf.divide(a,b)

  

tf.Tensor：id = 39，shape =（4，），dtype = float64，numpy = array（[1.，inf，1.，1。]）>

预期输出

  

tf.Tensor：id = 39，shape =（4，），dtype = float64，numpy = array（[1.，2.，1.，1。]）>

我见过this question，但它只处理单个值张量。可以以非常花哨的方式将它应用于具有多个元素的张量，但是没有其他实用的方法了吗？

PS：使用急切执行有什么不同。