Для себя Есть вот такой алгоритм (https://drive.google.com/file/d/1acnJA7DeTFK7XBAIHS57ndbHb9k0OQ1y/view?usp=sharing) Но сейчас в нем формируется граф на 49 вершин (строка с view). То есть, на выходе ResNet будет тензор 7х7х512, и мы делаем 49х512. Можно ли сделать так, что бы бралось не все 49, а как бы было среднее? Как я понял, для этого надо использовать AveragePooling. Если не ошибаюсь, должно получиться что-то вроде такого (часть с average) - https://drive.google.com/file/d/1LNGF1tSFvraysWwUzIGYDM_36Y3ehkby/view?usp=sharing