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