torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

将张量复制到固定内存中，如果它尚未被固定。默认情况下，设备固定内存将是当前加速器。