Thanks for your answer! Is that supported after some given version? Because I’m using 1.2.0 and NLLLoss only accepts rank-2 tensors, as described by @markl up above in this thread. If you try a higher rank tensor, you get the multi-target not supported error.