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 ·
157 阅读 ·
1 评论