Attention简单构造
attention简单构造
def build_model(class_num=2): # 二分类
inputs = Input(shape=(input_dim,)) #输入层
# 注意力层
attention_probs = Dense(input_dim, activation='softmax', name='attention_vec')(inputs)
attention_mul = Multiply()([inputs, attention_pro
原创
2021-03-29 23:31:51 ·
135 阅读 ·
1 评论