复制到

class dgl.graphbolt.CopyTo(datapipe, device)[source]

Bases: IterDataPipe

DataPipe 将从前一个 DataPipe 生成的每个元素传输到给定设备。对于 MiniBatch，默认情况下只会传输相关属性（自动推断）。

功能名称: copy_to.

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

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

  • datapipe (DataPipe) – 数据管道。

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