Pytorch - (Categorical) Cross Entropy Loss using one hot encoding and softmax

Question

I\'m looking for a cross entropy loss function in Pytorch that is like the CategoricalCrossEntropyLoss in Tensorflow.

My labels are one hot encoded and the p