预测结果

可以快速上手的开发文档

#预测结果

pytorch模型预测结果与ndarray互转方式

预测结果转为numpy:logits=model(feature)#如果模型是跑在GPU上result=logits.data.cpu().numpy()/logits.cpu()..