CopyTo

class dgl.graphbolt.CopyTo(datapipe, device, non_blocking=False)

基类: IterDataPipe

将前一个 DataPipe 生成的每个元素转移到指定设备的 DataPipe。对于 MiniBatch，默认只转移相关的属性（自动推断）。

函数名: copy_to。

当 data 实现了 to 方法时，CopyTo 将等同于

for data in datapipe:
    yield data.to(device)

参数:

• datapipe (DataPipe) – DataPipe。

• device (torch.device) – PyTorch CUDA 设备。

• non_blocking (bool) – 是否执行非阻塞复制。如果启用，所有元素必须已经在锁页系统内存中。默认为 False。