如何解决无法从tensorflow.keras.layers导入MultiHeadAttention的问题？

本文共计259个文字，预计阅读时间需要2分钟。

这里记录目录一、Keras的MultiHeadAttention导入API报错问题二、分析报错配置最终解决方案一、Keras的MultiHeadAttention导入API报错问题+这里记录目录+一、Keras的MultiHeadAttention导入API报错问题+二、分析报错配置最终解决方案

这里写目录标题一、keras的MultiHeadAttention导入API报错问题二、分析报错配置最终解决方案一、keras的MultiHeadAttention导入API报错问

这里写目录标题

• 一、keras的MultiHeadAttention导入API报错问题
• 二、分析报错配置
• 最终解决方案

一、keras的MultiHeadAttention导入API报错问题

无法引入MultiHeadAttention 导入包 from tensorflow.keras.layers import MultiHeadAttention

然后run后报错 cannot import name ‘MultiHeadAttention’ from tensorflow.keras.layers’

二、分析报错配置

查询keras官网 引入MultiHeadAttention发现报错在官网中给出了三种注意力机制层其他两种都能正常引入。尝试过tensorflow2.1和2.3都是缺少接口。

临时解决办法

pip install keras-multi-head

然而这是keras的方法我们更常用的还是tf.kerastensorflow2.0以上的。 估摸着是tensorflow版本等问题。一番尝试果不其然。

最终解决方案

只需要更换tensorflow2.5版本后导入即可成功( •̀ ω •́ )y 所以是tensorflow版本问题为了加快效率这里直接给出tensorflow2.5的安装配置可点击

配置项目版本python3.6-3.9TensorflowCPU/GPU2.5.0CudaGPU版本所需11.2CudnnGPU版本所需v 8.0.5