请问predict计算结果多出的一维具体是什么数值,还是只是一个填充

来源:4-4 利用神经网络解决分类和回归问题(2)

慕妹4365742

2020-06-13

写回答

2回答

一个很坏的好人

2020-06-13

个人理解

因为net的输出特征为1,所以pred的shape是(10,1)

假如net的输出特征为2,那么pred的shape将变成(10,2)

所以多出来这一维是有意义的

1
0

会写代码的好厨师

2020-06-27

分类问题计算结果一是时:N*C,其中C指的是类别的概率分布。

0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 298 问题

查看课程