如何将PyTorch模型预测输出的tensor转换为普通数值

操作方法 #

直接调用张量自带的item()方法即可提取单个元素张量的原生数值：

# output 为你得到的模型输出张量
result = output.item()

如果不需要保留梯度信息，建议先从计算图中剥离张量再提取，避免额外内存占用：

result = output.detach().item()

注意事项 #

• item()仅适用于只有1个元素的张量，若张量包含多个元素，可改用tolist()方法转为Python列表格式。
• 即使张量存储在GPU上，item()也会自动将数据拷贝到CPU后返回数值，无需额外调用cpu()方法。

内容的提问来源于stack exchange，提问作者user16668992