pytorch 'torch.round()'未按预期工作[duplicate]

gfttwv5a 于 2022-11-09 发布在其他

关注(0)|答案(1)|浏览(153)

此问题在此处已有答案：

Is floating point math broken?（31个答案）
19天前关闭。
我在GPU上运行一个深度网络，所以理想情况下，我的所有代码都应该在GPU下有效地工作（所以 Torch 而不是麻木），我试图用torch.round()舍入Tensor值，但结果不是预期的：

testing_value = torch.tensor(3.440000057220459)
torch.round(testing_value, decimals = 2)

这个输出结果是tensor(3.4400)，但它不应该是tensor(3.44)吗？
我该如何解决这个问题？

1条答案

代码应该如下所示：

testing_value = torch.tensor(3.440000057220459)
torch.set_printoptions(precision=2)
torch.round(testing_value, decimals = 2)

输出量：

tensor(3.44)