Good point. I tried to use PyTorch’s cross_entropy function but I got this error: "multi-target not supported".