Как читать пользовательские двоичные файлы в Tensorflow?

Question

У меня есть фрагмент кода для чтения некоторых двоичных файлов, которые записаны в указанном формате c. Я хочу написать тензорную версию моей read_file() функции. Есть ли способ сделать это? Я не хочу обернуть свою функцию, используя tf.py_function().

def read_file(filename):
    f = open(filename, 'rb')
    w = int(np.fromfile(f, np.int32, count=1))
    h = int(np.fromfile(f, np.int32, count=1))
    data = np.fromfile(f, np.float32, count=2 * w * h)
    data = np.resize(data, (h, w, 2))
    f.close()
    return data

Answer 1

К сожалению, каждый код python, который вы хотите использовать рядом с функциями tf *, должен быть заключен в tf.py_function().

Нет другого способа интеграции пользовательского кода Python с TensorFlow. специализированные функции.

Подробнее об этом можно узнать по этому вопросу: Есть ли альтернатива tf.py_function () для пользовательского Python кода?