GPU环境下torch.round()结果不符合预期，如何解决？

首先明确：tensor(3.4400)和tensor(3.44)的数值完全等价，差异仅来自PyTorch默认的打印格式。要让输出显示为3.44，可以通过以下两种方式处理：

1. 全局设置PyTorch打印精度 #

通过torch.set_printoptions()全局调整张量的打印小数位数，设置后所有张量的输出都会遵循这个格式：

import torch

# 设置全局打印精度为2位小数
torch.set_printoptions(precision=2)

testing_value = torch.tensor(3.440000057220459)
result = torch.round(testing_value, decimals=2)
print(result)  # 输出: tensor(3.44)

2. 单次打印时格式化输出 #

如果不想全局修改打印设置，可在打印时用Python的字符串格式化控制显示：

testing_value = torch.tensor(3.440000057220459)
result = torch.round(testing_value, decimals=2)
print(f"{result:.2f}")  # 输出: 3.44

注意：这种方式输出的是字符串，而非张量对象，适合仅需要查看结果的场景。

关键提示 #

• 无论显示格式如何，张量在GPU中的存储和计算逻辑完全不受影响，仍保持浮点数的精度特性，不会降低GPU运行效率。
• torch.round()的实际计算结果已经是正确的两位小数近似值，显示多出来的末尾零只是PyTorch默认打印格式导致的视觉问题。

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