pytorch学习笔记——BCE与CE

BCELoss的话只需要网络输出一个通道，CE Loss(Cross Entropy Loss)需要输出n_class个通道。
对于二分类任务可以使用CE Loss输出两个通道，也可以使用BCE Loss输出一个通道。
https://www.jianshu.com/p/5b01705368bb
https://zhuanlan.zhihu.com/p/372628727
https://zhuanlan.zhihu.com/p/580367698