Механизм внимания для cnn
Я работаю над EfficentNet-B0. Я хочу добавить в свою модель механизм внимания.
Результатом моего conv_layer является (None,7,7,1280), мне нужно преобразовать его в (49,1280), а затем использовать Ui как вход слоя FC. Я знаю, что регионы являются векторами (1,1280), поэтому для каждого изображения у меня есть 49 регионов.
Как мне это сделать? Спасибо!