TensorFlow函數(shù)教程：tf.io.TFRecordWriter

tf.io.TFRecordWriter函數(shù)

類 TFRecordWriter

別名：

• 類 tf.io.TFRecordWriter
• 類 tf.python_io.TFRecordWriter

定義在：tensorflow/python/lib/io/tf_record.py。

將記錄寫入TFRecords文件的類。

此類實(shí)現(xiàn)__enter__和__exit__，并且可以像普通文件一樣在with塊中使用。

__init__

__init__(
    path,
    options=None
)

打開path文件并創(chuàng)建一個(gè)寫入它的TFRecordWriter。

參數(shù)：

• path：TFRecords文件的路徑。
• options:（可選）指定壓縮類型，TFRecordCompressionType或TFRecordOptions對象的字符串 。
• 
  

可能引發(fā)的異常：

• IOError：如果無法打開path進(jìn)行寫入。
• ValueError：如果無法從options確定有效的compression_type。

方法

__enter__

__enter__()

輸入一個(gè)with塊 block。

__exit__

__exit__(
    unused_type,
    unused_value,
    unused_traceback
)

退出一個(gè)block，關(guān)閉文件。

close

close()

關(guān)閉文件。

flush

flush()

刷新文件。

write

write(record)

將字符串記錄寫入文件。

參數(shù)：

• record：str