Pytorch:正向传播推理(predict)出的outputs的值的每一个元素数为什么是两个或多个

两个或多个其实指的就是类别数,如果是一个二分类问题,就是两个。

红框和蓝框分别代表了每个类别的推理值,(此时可能通过了激活函数或者未通过),反正最后就是将通过激活函数后的值进行比较大小,大者为识别的类别,即predict对应的label 

posted @ 2023-02-17 11:01  龙雪  阅读(2)  评论(0编辑  收藏  举报  来源