When using only the last time step though, would’t there be the problem of the varying lengths of my data?
So i would need to use my original lengths, to get those positions respectively for each data in my batch.
And so far I didn’t figure out how to get this.
output[ : ,lengths-1]

was the first thing i tried, but it did not return the correct values. Am i just bad at indexing here?