Correct me if im wrong, but as far as I know, CE will output a loss of 0 when both of inputs are the same. Since CE can’t take the dimensions of my tensors as defined above, I am implementing a custom CE loss function that accepts said dimensions. Which is why I am expecting a zero.