Question on video loading

@JanuszL and team hope you're doing OK. I have a pytorch video model that assumes each batch element is a five-frame video sequence around a frame at index ind, i.e., [ind-2, ind-1, ind, ind+1, ind+2]. Now, I need to pass a batch of such sequences to that model, ie.,

seqs = []
for ind in range(batch_size):
    seq = # images [ind-2, ind-1, ind, ind+1, ind+2]
    seqs.append(seq)

I have a vision of how to do this in pytorch after getting the sequence from DALI. is there a simpler way to do this in DALI?