如何用mindspore的nn.SoftmaxCrossEntropyWithLogits算子完成pytorch中nn.CrossEntropyLoss算子的ignore_index功能

语义分割任务中计算交叉熵时通常会遇到ignore_index，这在pytorch中有非常简便的应用，但可惜mindspore中的交叉熵还不支持直接应用ignore_index，那我们在mindspore中怎么达到这一目的呢？

查看全文