https://cognitivehub.in/cross-entropy-in-pytorch/