PyTorch | Basics | Priority

Question 1

For a 2x2 outputs object, what will this code do? predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)

pytorch hugging-face

Answer 1

Softmax along the columns (here, the -1 dim) to produce probabilities across columns.

Source