TensorFlow函數(shù)教程：tf.io.TFRecordOptions

tf.io.TFRecordOptions函數(shù)

類 TFRecordOptions

別名：

• 類 tf.io.TFRecordOptions
• 類 tf.python_io.TFRecordOptions

定義在：tensorflow/python/lib/io/tf_record.py。

用于操作TFRecord文件的選項(xiàng)。

__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
)

創(chuàng)建一個(gè)TFRecordOptions實(shí)例。

當(dāng)compression_type不是None時(shí)，選項(xiàng)僅影響TFRecordWriter?？梢栽趜lib_compression_options.h和zlib手冊中找到文檔，詳細(xì)信息和默認(rèn)值。將選項(xiàng)保留為None，允許C ++設(shè)置合理的默認(rèn)值。

參數(shù)：

• compression_type：TFRecordCompressionType或None。
• flush_mode：flush模式或None，默認(rèn)值：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。默認(rèn)值：Z_DEFAULT_STRATEGY。

返回：

一個(gè)TFRecordOptions對象。

可能引發(fā)的異常：

• ValueError：如果compression_type無效。

方法

get_compression_type_string

@classmethod
get_compression_type_string(
    cls,
    options
)

將各種選項(xiàng)類型轉(zhuǎn)換為統(tǒng)一字符串。

參數(shù)：

• options：TFRecordOption，TFRecordCompressionType或字符串。

返回：

壓縮類型為字符串（例如'ZLIB'，'GZIP'或''）。

可能引發(fā)的異常：

• ValueError：如果compression_type無效。

類成員

compression_type_map