AttributeError:类型对象'DatasetV2‘没有特性'parallel_interleave’

Question

当我试图在tensorflow 2数据集上使用apply方法时，我遇到了这个错误，下面是我写的代码：

from tensorflow import data as tfdata

dataset = tfdata.Dataset.list_files(
    self._file_pattern,
    shuffle=self._is_training,
    seed=tf.random.set_random_seed(int(time.time() * 1e9)))
if self._is_training:
  dataset = dataset.repeat()

然后应用方法：

# Prefetch data from files.
def _prefetch_dataset(filename):
  dataset = tfdata.TFRecordDataset(
      filename, buffer_size=BUFFER_SIZE).prefetch(1)
  return dataset

dataset = dataset.apply(
    tfdata.Dataset.parallel_interleave(
    _prefetch_dataset, cycle_length=32, sloppy=self._is_training))

Answer 1

在TensorFlow2.x中，我们不再需要使用apply方法。只需使用dataset = dataset.interleave()而不是dataset.apply(tf.data.experimental.parallel_interleave())