Hi,

I have a problem where i need to calculate a simularity matrix before i can calculate my loss function. This simularity matrix takes a lot of time and is only dependant on the batch data itself. That is why i would like to calculate it in parrallel before the batch has started but i don’t know how i can do that. Is it possible with the torch.Dataloader? I know you can preprocess the data there and transform each sample on his own but can you aswell return a computation on the whole batch itself? Or is there another way to do this?