def METHOD_NAME(output: torch.Tensor) -> t.List[t.Tuple[str, t.List[float]]]:
    pred_value = output.argmax(-1).flatten().tolist()
    probability_matrix = torch.nn.Softmax(dim=1)(output).tolist()
    return list(zip([str(p) for p in pred_value], probability_matrix))