I see, so then you should change the shape of input_y to match these 10 tokens as well. You should consider the length of sequence in a different way, which really depends what model you want to use, like many-to-many, many-to-one, …