输出的标签是：[1., 0., 0., 0., 0.],
提取最大值所在的 index
labels_ = torch.max(labels, 1)[1]
### 返回最大值的索引
RuntimeError: 1D target tensor expected, multi-target not supported
猜你喜欢
转载自blog.csdn.net/weixin_42528089/article/details/104836539
今日推荐
周排行