在tf.keras中使用自定义CoordConv函数时，如何诊断张量对象没有对象“ is_initialized”？

Question

我使用以下仓库中的CoordinateChannel2D自定义层：https://github.com/titu1994/keras-coordconv，特别是来自coord.py文件。我得到张量对象没有对象'is_initialized'的错误。

x = CoordinateChannel2D(use_radius=True)(tiled_emb)

错误：

AttributeError: 'Tensor' object has no attribute 'is_initialized'

注意：原始回购使用keras。为了方便起见，我用他们的tf.keras替代了它们：

from tensorflow.keras.layers import Layer, InputSpec
from tensorflow.keras import backend as K
from tensorflow.keras.utils import get_custom_objects

代码在积极的执行模式下运行良好。但是，我真的更喜欢静态图，以实现更快的速度和多GPU的使用。