Hm, the problem I’m trying to solve is to load a batch in a multithreaded way, let’s say reading 500 images. In a normal dataset+dataloader setup it will happen sequentially (with prefetch optional) but I feel that reading those might be improved  this way as it’s a io heavy task.
But as I dig more into it might be less obvious and the TT ads  cpu overhead (transpose)
Anyhow thanks for taking the time to look into this, and if you have any suggestions lmk.