PyTorch打印输出为2dp

Question

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

例如

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

...在＆＃39; torch.set_printoptions＆＃39;但在这种情况下，我无法看到任何一个论点可能会有所帮助。

非常感谢提前！

Answer 1

现在已实施。使用

torch.set_printoptions(sci_mode=False)

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