当前位置:主页 > 软件编程 > Python代码 >

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

时间:2021-03-12 09:54:14 | 栏目:Python代码 | 点击:

预测结果转为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)

您可能感兴趣的文章:

相关文章