dgl.rand_graph

dgl.rand_graph(num_nodes, num_edges, idtype=torch.int64, device=device(type='cpu'))[源码]

生成具有给定节点数/边数的随机图并返回。

它从所有可能的节点对中均匀选择 num_edges 并形成一个图。随机选择是无放回的，这意味着结果图中不会有多条边。

要控制随机性，可以通过 dgl.seed() 设置随机种子。

参数：

  • num_nodes (int) – 节点数量

  • num_edges (int) – 边数量

  • idtype (int32, int64, 可选) – 用于存储结构相关图信息（如节点和边 ID）的数据类型。它应该是一个框架特定的数据类型对象（例如，torch.int32）。默认情况下，DGL 使用 int64。

  • device (设备上下文, 可选) – 结果图所在的设备。它应该是一个框架特定的设备对象（例如，torch.device）。默认情况下，DGL 将图存储在 CPU 上。

返回值：

生成的随机图。

返回类型：

DGLGraph

另请参阅

rand_bipartite

示例

>>> import dgl
>>> dgl.rand_graph(100, 10)
Graph(num_nodes=100, num_edges=10,
      ndata_schemes={}
      edata_schemes={})