TensorFlow函数教程：tf.io.TFRecordOptions

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