I have sequences of data created from multiple agents playing a game for a given amount of timesteps in synchron, thus with a shape of (#agents, #timesteps, x, …), where the data denoted by ‘x, …’ can be of any shape.
What would be an easy way to slice the ‘timestep’ long sequences of data into smaller sequential parts?
A specific example (with a picture as data):
From (32, 1024, 3, 28, 28) into (128, 256, 3, 28, 28), as we create 256 long sequences from 1024 long sequences and as a result increase our “number” of data from 32 to 4*32=128.
My only guess would be to make a fancy indexing such as one would do on a list of lists of lists, but there must be a better way.
Thanks is advance.