I think it should be possible.
In the example i provided earlier, lengths and x_length are the same value. And i’m trying to index it, but that doesn’t really work right now, as i seem to have made an error while indexing.
And another question i would have regarding this:
If i use this indexing, to get only my last time step (or mean over only the relevant time steps) would it still be possible, to do loss calculation and back propagation with the data?