torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

將 tensor 複製到鎖頁記憶體（如果尚未鎖定）。預設情況下，鎖頁記憶體所在的裝置將是當前的加速器。