Hi, thank you for your comment!
Then how can I prevent the mixing of element associated to different dataset inside a given batch?
And how to perform operation on a pair of batches (one for each dataset) at each step?