You can add it as a placeholder to indicate you don’t want to use this return value (the max. values) and only want to use the max. indices.
Alternatively, you could also directly use pred = torch.argmax(output, dim=1).