torch.Tensor.cpu

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

返回此对象在CPU内存中的副本。

如果此对象已经在CPU内存中并且位于正确的设备上，则不会执行复制操作，而是返回原始对象。

Parameters

memory_format (torch.memory_format, 可选) – 返回的张量所需的内存格式。默认值: torch.preserve_format。