Есть ли способ напечатать выходные данные оценки модели 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
... в разделе "torch.set_printoptions", но я не понимаю, как какой-либо из аргументов может помочь в этой ситуации.
Заранее большое спасибо!