阅读(10.7k) 书签 (0)

TensorFlow函数教程:tf.io.TFRecordWriter

2020-07-13 11:19 更新

tf.io.TFRecordWriter函数

类 TFRecordWriter

别名:

  • 类 tf.io.TFRecordWriter
  • 类 tf.python_io.TFRecordWriter

定义在:tensorflow/python/lib/io/tf_record.py。

将记录写入TFRecords文件的类。

此类实现__enter__和__exit__,并且可以像普通文件一样在with块中使用。


__init__

__init__(
    path,
    options=None
)

打开path文件并创建一个写入它的TFRecordWriter。

参数:


  • path:TFRecords文件的路径。
  • options:(可选)指定压缩类型,TFRecordCompressionType或TFRecordOptions对象的字符串 。

可能引发的异常:

  • IOError:如果无法打开path进行写入。
  • ValueError:如果无法从options确定有效的compression_type。

方法

__enter__

__enter__()

输入一个with块 block。



__exit__

__exit__(
    unused_type,
    unused_value,
    unused_traceback
)

退出一个block,关闭文件。




close

close()

关闭文件。



flush

flush()

刷新文件。



write

write(record)

将字符串记录写入文件。



参数:

  • record:str