TensorFlow函数教程:tf.io.TFRecordOptions
2019-02-26 16:52 更新
tf.io.TFRecordOptions函数
类 TFRecordOptions
别名:
- 类 tf.io.TFRecordOptions
- 类 tf.python_io.TFRecordOptions
定义在:tensorflow/python/lib/io/tf_record.py。
用于操作TFRecord文件的选项。
__init__
__init__(
compression_type=None,
flush_mode=None,
input_buffer_size=None,
output_buffer_size=None,
window_bits=None,
compression_level=None,
compression_method=None,
mem_level=None,
compression_strategy=None
)
创建一个TFRecordOptions实例。
当compression_type不是None时,选项仅影响TFRecordWriter。可以在zlib_compression_options.h和zlib手册中找到文档,详细信息和默认值。将选项保留为None,允许C ++设置合理的默认值。
参数:
- compression_type:TFRecordCompressionType或None。
- flush_mode:flush模式或None,默认值:Z_NO_FLUSH。
- input_buffer_size:int或None。
- output_buffer_size:int或None。
- window_bits:int或None。
- compression_level:0到9,或None。
- compression_method:压缩方法或None。
- mem_level:1到9,或None。
- compression_strategy:策略或None。默认值:Z_DEFAULT_STRATEGY。
返回:
一个TFRecordOptions对象。
可能引发的异常:
- ValueError:如果compression_type无效。
方法
get_compression_type_string
@classmethod
get_compression_type_string(
cls,
options
)
将各种选项类型转换为统一字符串。
参数:
- options:TFRecordOption,TFRecordCompressionType或字符串。
返回:
压缩类型为字符串(例如'ZLIB','GZIP'或'')。
可能引发的异常:
- ValueError:如果compression_type无效。
类成员
compression_type_map