train_tuner.py 文件源码

def tf_nan_to_zeros_float32(tensor):
    """
    Mask NaN values with zeros
    :param tensor that might have Nan values
    :return: tensor with replaced Nan values with zeros
    """
    return tf.select(tf.is_nan(tensor), tf.zeros(tf.shape(tensor), dtype=tf.float32), tensor)