Функция io.read_file

Функция io.read_file читает содержимое файла и возвращает его в виде скалярного тензора строкового типа. Первым параметром функция принимает путь к файлу или тензор с путем. Вторым необязательным параметром можно передать имя операции. Функция полезна при загрузке данных, например изображений или текстов, для дальнейшей обработки в графе TensorFlow.

Синтаксис

tf.io.read_file(filename, [name])

Пример

Давайте создадим временный файл и прочитаем его содержимое с помощью функции io.read_file:

import tensorflow as tf
import tempfile
import os

# Create a temporary file with content
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
	f.write('abcde')
	temp_path = f.name

t = tf.io.read_file(temp_path)
print(t)
print(t.numpy().decode('utf-8'))

# Clean up
os.remove(temp_path)

Результат выполнения кода:

tf.Tensor(b'abcde', shape=(), dtype=string)
"abcde"

Пример

Давайте прочитаем файл по пути, переданному в виде тензора, и выведем его размер:

import tensorflow as tf
import tempfile
import os

# Create a temporary file with content
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
	f.write('model.keras')
	temp_path = f.name

path_tensor = tf.constant(temp_path)
t = tf.io.read_file(path_tensor)

print(t)
print("File size:", len(t.numpy()))

# Clean up
os.remove(temp_path)

Результат выполнения кода:

tf.Tensor(b'model.keras', shape=(), dtype=string)
File size: 11

Смотрите также

функцию write_file,
которая записывает данные в файл
функцию matching_files,
которая возвращает список файлов, соответствующих шаблону
функцию decode_base64,
которая декодирует строки в формате Base64
функцию encode_base64,
которая кодирует строки в формат Base64