Solve cannot import name 'multiheadattention' from 'tensorflow keras. layers‘

One 、keras Of MultiHeadAttention Import API Report the wrong question

Can't introduce MultiHeadAttention

Import package ：
from tensorflow.keras.layers import MultiHeadAttention

then run Report a mistake later
cannot import name ‘MultiHeadAttention’ from 'tensorflow.keras.layers’

Two 、 Analyze the error reporting configuration

Inquire about keras Official website

introduce MultiHeadAttention, Found an error , Three attention mechanism layers are given in the official website , The other two can be introduced normally . Tried tensorflow2.1 and 2.3, All lack interfaces .

A temporary solution

pip install keras-multi-head

However, this is keras Methods , What we use more often is tf.keras（tensorflow2.0 The above ）.
Estimated yes tensorflow Version and other issues . Try , Sure enough .

Final solution

Just replace tensorflow2.5 You can import the version successfully ( •̀ ω •́ )y
So it is tensorflow Version of the problem , In order to speed up efficiency , It is directly given here tensorflow2.5 Installation configuration （ Clickable ）

Configuration items	edition
python	3.6-3.9
Tensorflow（CPU/GPU）	2.5.0
Cuda（GPU Version required ）	11.2
Cudnn（GPU Version required ）	v 8.0.5