I am writing a custom module. I want to compute the batch statistics (like running mean in Batch Normalization) by the main GPU and then distributed to other GPUs. How to share a value across different GPUs?