将tensforflow中的所有nan值转换为零

我正在尝试转换所有 nan 最终结果中的值为零。我无法正确执行它!

代码如下:也可在colab上获得: link

import tensorflow as tf
import numpy as np

ts = tf.constant([[0,0]]) 
tx = tf.constant([[0,1]]) 

out = ts / (ts + fx)

out.numpy() # array([nan,  0.])

tf.math.is_nan(out).numpy() # array([ True, False]

out.numpy()[(tf.math.is_nan(out).numpy())] = 0
out.numpy() #array([nan,  0.])

out.numpy() 应该给 array([0., 0.])