Python tf.data.Dataset.with_options用法及代碼示例

用法

with_options(
    options, name=None
)

返回具有給定選項集的新 tf.data.Dataset。

這些選項是"global"，因為它們適用於整個數據集。如果選項設置了多次，隻要不同的選項不使用不同的非默認值，它們就會被合並。

ds = tf.data.Dataset.range(5)
ds = ds.interleave(lambda x:tf.data.Dataset.range(5),
                   cycle_length=3,
                   num_parallel_calls=3)
options = tf.data.Options()
# This will make the interleave order non-deterministic.
options.deterministic = False
ds = ds.with_options(options)