I was trying to make a cross entropy weighted on pytorch following with the normal cross entropy, but I don\'t know how to insert the weight in it. The equation is: