当前位置：网站首页>torch.nn.CrossEntropyLoss()对应的MindSpore算子是哪个？

torch.nn.CrossEntropyLoss()对应的MindSpore算子是哪个？

2022-08-10 03:28:00 【小乐快乐】

是和SoftmaxCrossEntropyWithLogits对应吗？能否完全对应？

不能完全对应，MindSpore的SoftmaxCrossEntropyWithLogits没有weight，其他都一样

原网站

版权声明
本文为[小乐快乐]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_45666880/article/details/126249437

边栏推荐

猜你喜欢

随机推荐