keras注意力机制 cnn图像

def cbam(inputs):
    inputs_channels=int(inputs.shape[-1])
    x=GlobalAveragePooling2D()(inputs)
    x=Dense(int(inputs_channels/4))(x)
    x=Activation(relu)(x)
    x=Dense(int(inputs_channels))(x)
    x=Activation(softmax)(x)
    x=Reshape((1,1,inputs_channels))(x)
    x=Multply()([inputs,x])
    return x