采纳答案成功!
向帮助你的同学说点啥吧!感谢那些助人为乐的人
个人理解
因为net的输出特征为1,所以pred的shape是(10,1)
假如net的输出特征为2,那么pred的shape将变成(10,2)
所以多出来这一维是有意义的
分类问题计算结果一是时:N*C,其中C指的是类别的概率分布。
登录后可查看更多问答,登录/注册
理论基础+技术讲解+实战开发,快速掌握PyTorch框架
1.7k 7
1.5k 7
1.6k 6
969 5
2.5k 5
购课补贴联系客服咨询优惠详情
慕课网APP您的移动学习伙伴
扫描二维码关注慕课网微信公众号