Linking softmax probabilities to classes in a multi-class ...
discuss.pytorch.org › t › linking-softmaxAug 19, 2020 · I have a multi-class problem, the classes are all encoded 0-72. I have an preds tensor of [256, 72]. Passing it through probs = torch.nn.functional(input, dim = 1) results in a tensor with the same dimensionality. Where probs[0] is a list of probabilities of each class being the correct prediction. I would like to analyse the predictions my model is making, how can I link the probabilities to ...