def _tf_safe_reciprocal(x):
return tf.reciprocal(x + tf.cast(tf.equal(x, 0), x.dtype))
微信
公众号
扫码关注公众号
微信
公众号