python - PyTorch 打印输出到 2dp

Question

如果输出是多元素 FloatTensor，是否有办法将 CNN 模型评估的输出打印为 2dp？

例如。

prediction = torch.exp(model(image2))
print(prediction)

输出:

Variable containing:
 2.84e-01  1.68e-07  7.16e-01
[torch.FloatTensor of size 1x3]

如果我可以将值输出为:

Variable containing:
 0.28  0.00  0.72
[torch.FloatTensor of size 1x3]

我已经尝试过:

print("%.2f" % prediction)

并使用:

torch.set_printoptions(precision=2)

但都没有达到预期的效果。

我查看了文档页面:

http://pytorch.org/docs/master/torch.html#creation-ops

...在“torch.set_printoptions”下，但我看不出任何参数在这种情况下有什么帮助。

非常感谢!

Answer 1

现在已经实现了。使用

torch.set_printoptions(sci_mode=False)

https://pytorch.org/docs/stable/torch.html#torch.set_printoptions