如何获取PyTorch中Tensor对象的具体数值？

你拿到的是一个位于CUDA设备上的PyTorch单元素张量，提取其中具体数值可以用以下两种方法：

• 先将张量转移到CPU，再调用item()方法（仅适用于单元素张量）：

  # 假设你的张量变量名为tensor_val
  num = tensor_val.cpu().item()
  

• 直接调用tolist()方法（同样适用于单元素张量，无需手动转移设备）：

  num = tensor_val.tolist()
  

注意：如果是多元素张量，tolist()会返回对应数值组成的列表，而item()仅支持单元素张量，使用时会报错。

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