如何编写高纬度输入的NLLLloss接口，——将d1, d2, …, dK维度拆分计算loss后再拼接的NLLLoss接口

请问这个损失进行迁移的时候如何进行封装拆分计算

可通过以下方式解决。

第一步，将 input (N, C, d1, d2, ..., dk) 转置（transpose）为 (N, d1, d2, ..., dk, C)

第二步，将 (N, d1, d2, ..., dk, C) reshape为 (N * d1* ... *dk, C)

第三步，将labels reshape为(N * d1* ... *dk, C)

第四步，使用MindSpore的NLLLoss，此时就是2维场景。