How to get the partial derivative of the output data with the middle layer data

[jpeg729]

Presumably label contains the labels as class identifier numbers, not as a 1-hot encoding.

Convert int into one-hot format

Hi, You can use the scatter_ method to achieve this. I would also advise to create the y_onehot tensor once and then just fill it: import torch batch_size = 5 nb_digits = 10 # Dummy input that HAS to be 2D for the scatter (you can use view(-1,1) if needed) y = torch.LongTensor(batch_size,1).random_() % nb_digits # One hot encoding buffer that you create out of the loop and just keep reusing y_onehot = torch.FloatTensor(batch_size, nb_digits) # In your for loop y_onehot.zero_() y_onehot.scat…