Overview of location tensors and models in Multi-GPU context

Is there a way to get an overview of which tensors and models are on which GPUs?
Ideally would be a function that returns a list of tensors/models given the cuda device index.

It would be very handy when debugging multi-GPU training scripts.

Thanks in advance,