The sum of the whole tensor is 1. what i want is the sum should be for each P in the tensor (p,1,s,s).
a softmax that normalize a 2D tenor in its spatial values. not in their channels and not in their batch