不受支持的操作解决方法

Question

我有一个自定义层，它使用 tf.py_function 和一些 python 代码。由于 Cloud TPU 不支持这一点，是否可以将此层的计算放在 CPU 或 GPU 设备上？或者是重写函数的唯一解决方案？

Answer 1

目前不支持此用例。