当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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


表示数据集线程的选项。

用法

tf.data.ThreadingOptions()

属性

  • max_intra_op_parallelism 如果设置，它将覆盖 intra-op 并行度的最大程度。
  • private_threadpool_size 如果设置，数据集将使用给定大小的私有线程池。值 0 可用于指示线程池大小应在运行时根据可用 CPU 内核的数量确定。

您可以通过 tf.data.Options 的 experimental_threading 属性设置数据集的线程选项；该属性是 tf.data.ThreadingOptions 的一个实例。

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

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.ThreadingOptions。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。