torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

如果张量尚未固定，则将其复制到固定内存。默认情况下，固定内存将在当前的 加速器 上。