Thank you for taking the time to reply @Topsoil.
My problem is actually a text classification problem and I want to add the category of that text as an additional input to my cnn to help it remember from which parent category the text is coming. I thought of adding one hot vector that has all the categories but I’m just getting started with pytorch so I dont know how to do that code wise.