dgl.rand_graph

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

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

它从所有可能的节点对中均匀选择num_edges并形成一个图。 随机选择是不放回的,这意味着在生成的图中不会有重复的边。

为了控制随机性,通过dgl.seed()设置随机种子。

Parameters:
  • num_nodes (int) – 节点的数量

  • num_edges (int) – 边的数量

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

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

Returns:

生成的随机图。

Return type:

DGLGraph

另请参阅

rand_bipartite

示例

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