Just curious whether the softmax in a Fully Convolutional Network equals to Softmax2D ? As in the source code it seems to be true. But the current version of Pytorch document does not mention anything about 2D softmax.
i think it is just missing in documentation.
Agreed. Adding this to the document will reduce ambiguities.
added in now. sorry for the delay.