Метод write класса TFRecordWriter

Метод write класса TFRecordWriter записывает одну запись в файл формата TFRecord. Первым параметром метод принимает сериализованную строку байтов, которую необходимо записать. Запись происходит последовательно, каждая новая запись добавляется в конец файла. Метод не возвращает значения, но может вызвать исключение при ошибке записи.

Синтаксис

tf.io.TFRecordWriter.write(record)

Пример

Давайте создадим объект TFRecordWriter и запишем одну строку в файл:

import tensorflow as tf

writer = tf.io.TFRecordWriter('data.tfrecord')
writer.write(b'abcde')
writer.close()

print("record written")

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

"record written"

Пример

Давайте запишем несколько записей в файл TFRecord:

import tensorflow as tf

writer = tf.io.TFRecordWriter('data.tfrecord')
writer.write(b'first record')
writer.write(b'second record')
writer.write(b'third record')
writer.close()

print("records written")

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

"records written"

Пример

Давайте сериализуем тензор и запишем его в файл TFRecord:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
serialized = tf.io.serialize_tensor(t)

writer = tf.io.TFRecordWriter('data.tfrecord')
writer.write(serialized.numpy())
writer.close()

print("tensor written")

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

"tensor written"

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

класс TFRecordWriter,
который создает объект для записи в файл TFRecord
метод close,
который закрывает файл TFRecord
метод flush,
который сбрасывает буфер в файл TFRecord
метод write,
который записывает данные в файл TFRecord