torch.Tensor.cpu

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

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

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

Параметры:

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