Solved by LeviViana in post #2 
              
              
                You could try the Kullback-Leibler divergence Loss. Here is the PyTorch documentation for this loss function. I’m assuming you are not dealing with a classification problem (i.e. the target or ground-truth probabilities are not one-hot encoded), otherwise, you could use a Cross Entropy loss.