Tensorflow 1.X раньше имел слой tensorflow.contrib.layers.spatial_softmax. Слой в основном делает softmax по каждому каналу и возвращает координаты максимальной точки.
Однако, эта функциональность отсутствует в TF 2.0. Это просто еще не портировано, или я что-то пропустил? Есть ли альтернативы?