每次迭代删除张量流数据集图

Question

我需要多次迭代来操作张量流数据集。但是，当使用地图操作数据集时，我遇到了内存问题。我非常确定张量流每次调用映射时都会绘制函数图，并且我想每次迭代删除该图。

我正在尝试使用 tf.keras.backend.clear_session 并手动删除它，但我仍然遇到内存问题。

for i in range(10000):
  keras.backend.clear_session()
  ds = tf_dataset.map(map_fn)
  del ds