libtorch: Why am I getting “c10::Error at memory location 0x000000C5539CC010.” when calling torch::nll_loss()?

Question

Update: I added a Minimal Reproducible Example below the original question.

I\'ve been debugging this error for some time now. It occurs when I call, t