weixin_44577815的博客

原创 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