The pytorch tag starts at 0. How to make the tag input according to its own category. For example, this type of data is 3, and the other class is 4, instead of starting from 0.I want to do this because the network output does not use softmax, but instead outputs the result of linear regression, compared to my own category label. thanks.