torch.Tensor.cpu

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

Returns a copy of this object in CPU memory.

If this object is already in CPU memory, then no copy is performed and the original object is returned.

See Device-to-host copies on NVLink-C2C systems for guidance on repeated CUDA-to-CPU transfers and reusing CPU buffers.

Parameters:

memory_format (torch.memory_format, optional) – the desired memory format of returned Tensor. Default: torch.preserve_format.