报错信息:
在用CrossEntropyLoss()损失函数时报错:
Dimension out of range (expected to be in range of [-1, 0], but got 1)
原因:
标签和输出矩阵不对
- 标签应为1维数据(不用转为one-hot向量),为long类型
- 输出矩阵为2维数据,为float类型
报错信息:
在用CrossEntropyLoss()损失函数时报错:
Dimension out of range (expected to be in range of [-1, 0], but got 1)
原因:
标签和输出矩阵不对