torch.Tensor.copy_

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

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

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

Параметры

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