@ptrblck  I am sorry If I did not understand your question.
I chose [3] as a label out of 10 labels.
Usually, I will have labels shape as [Batch, label] but yes in this case label=[3] (shape -> [1,10] ).
The label is just an indicator for preforming one-hot masking on ‘X’.