tf.math.reciprocal_no_nan