torch.Tensor.copy_

Tensor.copy_(src, non_blocking=False) → Tensor

Копирует элементы из src в тензор self и возвращает self.

Тензор src должен быть совместим по правилам трансляции с тензором self. Он может иметь другой тип данных или находиться на другом устройстве.

Параметры:

• src (Tensor) – исходный тензор для копирования
• non_blocking (bool) – если True и эта копия выполняется между CPU и GPU, то копирование может происходить асинхронно по отношению к хосту. В других случаях этот аргумент не влияет.