pytorch中模型预测结果与ndarray互转的示例分析

预测结果转为numpy：

logits=model(feature) 
#如果模型是跑在GPU上result=logits.data.cpu().numpy()   /   logits.cpu().numpy() 
#如果模型跑在cpu上result=logits.data.numpy()   /   logits.numpy()

将矩阵转为tensor：

np_arr = np.array([1,2,3,4])tensor=torch.from_numpy(np_arr)

以上是“pytorch中模型预测结果与ndarray互转的示例分析”这篇文章的所有内容，感谢各位的阅读！相信大家都有了一定的了解，希望分享的内容对大家有所帮助，如果还想学习更多知识，欢迎关注亿速云行业资讯频道！