torch.Tensor.copy_

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

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

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

Параметры:

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

Примечание

Если non_blocking равно True и копирование выполняется в потоке CUDA, отличном от потока по умолчанию, вызывающий код отвечает за правильную синхронизацию между потоками. Описание необходимого шаблона см. в разделе потоки CUDA.