torch.utils.swap_tensors

torch.utils.swap_tensors(t1, t2) [исходный код]

Эта функция меняет местами содержимое двух объектов Tensor. В общих чертах, она делает так, чтобы t1 содержал данные t2, сохраняя при этом его идентичность.

Это не сработает, если у t1 и t2 разные слоты.