hi,
I don’t understand much，Why is the kl_div() input and target sum must be equal to 0, the sum on the entire feature map is  0, or the sum of a pixel in the feature map in the different channels is 0 ?
thanks