tensorflow - 无法在 Keras 中导入 AttentionLayer (TF1.13)

Question

我正在尝试为我的编码器解码器模型导入注意力层，但它给出了错误。

from keras.layers import AttentionLayer

或者

from keras.layers import Attention

以下是错误
cannot import name 'AttentionLayer' from 'keras.layers'
cannot import name 'Attention' from 'keras.layers'
任何建议？

---

标签： tensorflowkerasattention-model

Answer 1

我通过升级到 tensorflow 1.14 并将其导入解决了这个问题

from tensorflow.keras.layers import Attention