torch.Tensor.pin_memory

Tensor.pin_memory() → Tensor

Копирует тензор в закреплённую память, если он ещё не закреплён. По умолчанию устройством, на котором закрепляется память, будет текущий ускоритель.