dgl.save_graphs

dgl.save_graphs(filename, g_list, labels=None, formats=None)[source]

将图形及其标签（可选）保存到文件。

除了保存到本地文件，DGL 还支持直接将图写入 S3（通过提供 "s3://..." 路径）或 HDFS（通过提供 "hdfs://..." 路径）。

该函数将图结构和节点/边特征保存到文件中，使用DGL自己的二进制格式。对于图级别的特征，通过labels参数传递它们。

Parameters:

filename (str) – 用于存储图形和标签的文件名。
g_list (list) – 要保存的图形。
labels (dict[str, Tensor]) – labels 应该是张量的字典，键为字符串
格式 (str 或 list[str]) – 以指定格式保存图形。可以是 coo, csc 和 csr 的任意组合。如果未指定，则根据可用的格式仅保存一种格式。如果有多种格式可用，选择的优先级从高到低为 coo, csc, csr。

示例

>>> import dgl
>>> import torch as th

创建DGLGraph对象并初始化节点和边的特征。

>>> g1 = dgl.graph(([0, 1, 2], [1, 2, 3]))
>>> g2 = dgl.graph(([0, 2], [2, 3]))
>>> g2.edata["e"] = th.ones(2, 4)

将图形保存到文件中

>>> from dgl.data.utils import save_graphs
>>> graph_labels = {"glabel": th.tensor([0, 1])}
>>> save_graphs("./data.bin", [g1, g2], graph_labels)

另请参阅