Assert input.dim() == 2, 'Softmax requires a 2D tensor as input' AssertionError: Softmax requires a 2D tensor as input

Hello,

what’s wrong with my model

    m=torch.nn.Softmax()
    model.eval()
    preds = model(image)
    temps=preds.cpu()
    prob=torch.max(m(temps)*100)

error with prob variable
assert input.dim() == 2, 'Softmax requires a 2D tensor as input’
AssertionError: Softmax requires a 2D tensor as input