这里写目录标题一、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