253 字 1 分钟
<!-- moere --> # PyG 概述 # PyG 中的数据对象 PyG 中的数据对象即 torch_geometric.data.Data ,它是一个包含节点特征、边、边特征、图标签等信息的容器。 # PyG Batch PyG Batch 的创建需要使用 collate_fn 函数。该函数将多个 Data 对象合并为一个 Batch 对象。需要注意的是,如果需要重写 collect_fn, 那么需要使用 torch.utils.data 中的 DataLoader ,而不是 torch_geometric.data.DataLoader 。
more...