Убедитесь, что файл существует внутри потока Tensorflow.Использование tf.gfile.Exists со строкой Tensor в качестве входных данных

Question

Использование Tensorflow Я пытаюсь проверить, существует ли файл, прежде чем читать его с помощью tf.read_file(filename).К сожалению, при настройке моего конвейера я генерирую строку имени файла на лету с помощью команд tf.Я генерирую строку имени файла, используя tf.string_join, а затем хотел бы проверить, существует ли файл, вызвав tf.gfile.Exists.К сожалению, tf.gfile.Exists принимает только строки, но не Tensors.Как я могу проверить, что файл существует в Tensorflow?Могу ли я оценить тензор во время выполнения?Любой другой обходной путь или правильный способ сделать это?

Answer 1

Я просто завернул его в tf.py_func

def file_exists(file_path):
    [exists] = tf.py_func(_file_exists, [file_path], [tf.bool])
    exists.set_shape([])

    return exists

def _file_exists(file_path):
    return tf.gfile.Exists(file_path)