TensorFlow函数教程:tf.io.TFRecordWriter

由 Carrie 创建, 最后一次修改 2019-02-25

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块。

__exit__

__exit__(
    unused_type,
    unused_value,
    unused_traceback
)

退出一个with块,关闭文件。

close

close()

关闭文件。

flush

flush()

刷新文件。

write

write(record)

将字符串记录写入文件。

参数:

  • record:str
以上内容是否对您有帮助:
二维码
建议反馈
二维码