logits_per_image = outputs.logits_per_image probs = logits_per_image.softmax(dim=1) >>> probs tensor([[0.9949, 0.0051]])