torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

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