仅作为记录，大佬请跳过。
用
q=torch.argmax(pred_target, 1) list(q.cpu().detach().numpy())
参考
其中.detach()好像是为了不再有梯度——传送门
![将网络输出的pred，从tensor[[0.1,0.9],[0.4,0.6]]变成[1,1]——.cpu().detach().numpy()](http://www.mshxw.com/aiimages/31/876233.png "将网络输出的pred，从tensor[[0.1,0.9],[0.4,0.6]]变成[1,1]——.cpu().detach().numpy()")
仅作为记录，大佬请跳过。
用
q=torch.argmax(pred_target, 1) list(q.cpu().detach().numpy())
参考
其中.detach()好像是为了不再有梯度——传送门
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
