great! can I have 1 last question?
Let’s say I want to feed the kl-loss with an image and a target image, so he is doing the kl_div on each pixel? (and than so sum,mean etc…that I understood).
Just to be sure that I understood correctly…