pin_memory_fn

torchdata.datapipes.utils.pin_memory_fn(data, device=None)

将数据移动到固定内存的实用函数。如果需要特殊处理才能将输入数据移动到固定内存，请将一个 pin_memory 方法附加到预期数据类。