Spatial softmax

Greetings,

Spatial softmax layer is used to find the expected pixel location of feature maps, TensorFlow has an implementation for it. But I cannot find Torch’s one.

Could someone please help me with this?

FYI, here Is the link of the paper first introduced this concept. https://arxiv.org/abs/1504.00702.