Tensorflow 得到了张量流中的二阶梯度计算问题

最近我正在执行梯度惩罚Wasserstein GAN。我有一个问题：

ValueError:名称中没有名为“\u XLA编译”的属性： “渐变/模型2/展平1/重塑梯度/重塑”op:“重塑” 输入：“梯度/模型2\u 2/密集2/MatMul\u梯度/MatMul”输入： “渐变/模型2/展平1/重塑梯度/形状”设备： “/device:GPU:0”attr{key:“T”值{ 类型：DT_FLOAT}attr{key:“Tshape”值{ 类型：DT_INT32}

---

我觉得梯度计算的这一部分不知怎么的缺失了，或者我在TensorFlow图中使用了重塑。有人有类似的问题吗？

你想用XLA吗？也许去掉它，看看它是否还能用。更多的上下文围绕代码也会有帮助。我不想使用XLA…不知怎的，我犯了这样的错误。但我用tf.slice替换了keras裁剪层，解决了我的问题。（我混合了keras和tensorflow来构建我的模型）顺便说一句，谢谢！