TensorFlow2.0中的spatial_softmax层

Question

Tensorflow 1.X曾经有一个层tensorflow.contrib.layers.spatial_softmax。该层基本上在每个通道上执行softmax，并返回最大点的坐标。

但是，在TF 2.0中似乎缺少此功能。这是不是还没有移植，或者我错过了什么？还有别的选择吗？

Answer 1

查看this评论，显然空间softmax可以很容易地实现。