torch.Tensor.cpu

Tensor.cpu(memory_format=torch.preserve_format) → Tensor

Возвращает копию этого объекта в памяти CPU.

Если этот объект уже находится в памяти CPU, копирование не выполняется и возвращается исходный объект.

Параметры:

memory_format (torch.memory_format, необязательный) – желаемый формат памяти возвращаемого тензора. По умолчанию: torch.preserve_format.