nll_loss_forward_reduce_cuda_kernel_2d_index not implemented for Int 报错,类别标签需要转为

遇到 nll_loss_forward_reduce_cuda_kernel_2d_index not implemented for Int 报错,核心原因是送入 NLLLoss 的标签张量数据类型