Multinomial cross entropy loss pytorch. Hi , I have a binary segmentation problem.