torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

如果张量尚未位于固定内存中，则将其复制到固定内存。默认情况下，固定内存所在的设备是当前的 加速器。