當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.data.ThreadingOptions用法及代碼示例


表示數據集線程的選項。

用法

tf.data.ThreadingOptions()

屬性

  • max_intra_op_parallelism 如果設置,它將覆蓋 intra-op 並行度的最大程度。
  • private_threadpool_size 如果設置,數據集將使用給定大小的私有線程池。值 0 可用於指示線程池大小應在運行時根據可用 CPU 內核的數量確定。

您可以通過 tf.data.Optionsexperimental_threading 屬性設置數據集的線程選項;該屬性是 tf.data.ThreadingOptions 的一個實例。

options = tf.data.Options()
options.threading.private_threadpool_size = 10
dataset = dataset.with_options(options)

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.ThreadingOptions。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。