Hi.Thanks for your reply!
I want to create a model that gives me more than 10 output matrices and want to speed up the whole process by running multiple MSELoss functions at once on GPU instead of calling them in a for loop. With the normal usage, my GPU-Load is too low, which is just waste of time
With nonlocal out, child processes are actually supposed to use the main out of paral_MSELoss(). I also tried with the more powerful global out(which is actually wrong in my opinion) and that doesn’t work as well.