python - 在 TF(2.0) 模型中使用循环缓冲区

Question

我正在寻找一种方法来优化流数据的 tensorflow (2.0) UNet 模型。目标是将先前预测的激活存储在循环缓冲区中，以便在下一个预测中重用部分缓冲区。

我发现了以下 TensorFlow op FIFOQueue，从概念上讲，这是正确的方向。但是，它似乎只用于数据加载的上下文中。.layers我在文档中找不到任何好的线索。

这可以用 Tensorflow API 解决吗？这个问题是否足够低级以打破进入 C++ API 的障碍？目前我正在使用 Python 进行开发。

标签： pythontensorflowcircular-buffer