将网络输出的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()好像是为了不再有梯度——传送门