# 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 。