The error statement is as follows.
Is it working only when the index is 0?
Could you just tell me?
grads['dl/df'][y] -= 1
IndexError: index 1 is out of bounds for dimension 0 with size 1

print(y.size())
torch.Size([19573, 6])