I tested both
giving an input x of size (1,1,s,s)
nn.softmax(X[0,0]) 
nn.softmax2d(x)

the sum of the array or the feature map in the first is 100 while in the second is 1000000
the maximum value is 1 but it seems like it doesn’t apply softmax on all the array items ensemble