Python tf.compat.v1.data.TFRecordDataset用法及代碼示例

參數

• filenames 一個 tf.string 張量或 tf.data.Dataset 包含一個或多個文件名。
• compression_type (可選。)tf.string 標量評估為 ""(無壓縮)、"ZLIB" 或 "GZIP" 之一。
• buffer_size (可選。)tf.int64 標量，表示讀取緩衝區中的字節數。如果您的輸入管道存在 I/O 瓶頸，請考慮將此參數設置為 1-100 MB 的值。如果 None ，則使用本地和遠程文件係統的合理默認值。
• num_parallel_reads (可選。)tf.int64 標量，表示要並行讀取的文件數。如果大於1，則以交錯的順序輸出並行讀取的文件記錄。如果您的輸入管道存在 I/O 瓶頸，請考慮將此參數設置為大於 1 的值以並行化 I/O。如果 None ，文件將被順序讀取。
• name (可選。) tf.data 操作的名稱。

拋出

• TypeError 如果任何參數不具有預期的類型。
• ValueError 如果任何參數不具有預期的形狀。

屬性

• element_spec 此數據集元素的類型規範。

  dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
  dataset.element_spec
  TensorSpec(shape=(), dtype=tf.int32, name=None)

  如需更多信息，請閱讀本指南。

• output_classes 返回此數據集元素的每個組件的類。 (已棄用)

  警告：此函數已棄用。它將在未來的版本中刪除。更新說明：使用 tf.compat.v1.data.get_output_classes(dataset) 。

• output_shapes 返回此數據集元素的每個組件的形狀。 (已棄用)

  警告：此函數已棄用。它將在未來的版本中刪除。更新說明：使用 tf.compat.v1.data.get_output_shapes(dataset) 。

• output_types 返回此數據集元素的每個組件的類型。 (已棄用)

  警告：此函數已棄用。它將在未來的版本中刪除。更新說明：使用 tf.compat.v1.data.get_output_types(dataset) 。