So you mean, if originally it is 5x11x4 and 5x1x4 is the targets. I can have 20x11 and 20x1 and then if i have the first sentence of 3 words and second of 4 , select indices 0,1,2,4,5,6,7 from both vectors [These need to be calculated based on sentence lengths ]and apply the loss function ? Thanks a lot !