Как получить вероятности при использовании densenet Pytorch?

Question

Я хочу сделать двоичную классификацию, и я использовал DenseNet от Pytorch.

Вот мой код предсказания:

densenet = torch.load(model_path)
densenet.eval()
output = densenet(input)
print(output)

И вот результат:

Variable containing:
54.4869 -54.3721
[torch.cuda.FloatTensor of size 1x2 (GPU 0)]

Я хочу получить вероятности каждого класса.Что я должен делать?

Я заметил, что torch.nn.Softmax() можно использовать, когда есть много категорий, как обсуждено здесь .