如何使用keras构建注意力模型？

from keras.layers import Input, Dense, Activation, Flatten, Permute, Multiply
from keras.layers import LSTM, Bidirectional, Dropout
from keras.models import Model

input_dim = 100  # 输入维度
output_dim = 10  # 输出维度

input = Input(shape=(input_dim,))
x = Dense(64, activation='tanh')(input)
x = Dropout(0.5)(x)
x = Dense(32, activation='tanh')(x)
x = Dropout(0.5)(x)
x = Dense(output_dim, activation='softmax')(x)

attention_probs = Dense(1, activation='softmax')(x)
attention_mul = Multiply()([x, attention_probs])

model = Model(inputs=input, outputs=attention_mul)

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(X_train, y_train, epochs=10, batch_size=32)