Можно ли переопределить метод 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.])>
Я видел этот вопрос , но он обращается только к тензорам с одним значением.Вероятно, он может быть применен очень хитроумным способом к тензору с более чем одним элементом, но разве нет более практичного способа?
PS: Использование нетерпеливого исполнения, если оно имеет какое-либо значение.