我在哪里可以确定哪些 Keras 方法是 TPU 兼容方法？

Question

我最近了解到 UpSampling2D 不是 TPU 支持的操作。

如何找出哪些 Keras 函数与 Colab 中的 TPU 兼容？我找不到有关该主题的任何文档。

Answer 1

通常应该支持所有 Keras 层，但没有一个很好的总结。有一个更笼统的总结与 TPU Ops here 相关。

特别是，UpSampling2D 目前正在进行中，但如果您在使用 Keras 和 TPU 时遇到更多问题，我们鼓励您提交问题 here。