mindspore_gl.graph.BatchMeta

查看源文件
class mindspore_gl.graph.BatchMeta(graph_nodes, graph_edges)[源代码]

BatchMeta,批处理图形的元信息。

参数:
  • graph_nodes (numpy.array) - 批处理图中图的累积节点和的数组(第一个元素为0)。

  • graph_edges (numpy.array) - 批处理图中图的累积边和的数组(第一个元素为0)。

支持平台:

Ascend GPU

样例:

>>> from mindspore_gl.graph import BatchMeta
>>> graph_nodes = np.array([0, 20, 40, 60, 80])
>>> graph_edges = np.array([0, 100, 200, 300, 400])
>>> graph = BatchMeta(graph_nodes, graph_edges)
>>> print(graph[1])
(20, 100)
property edge_map_idx

边的索引列表。

返回:

numpy.array,数组,指示每个边的图索引。

property graph_count

图的数量。

返回:

int,此批处理图中的总图计数。

property graph_edges

图的边数组。

返回:

numpy.array,批处理图中图的累积边和(第一个元素为0)。

property graph_nodes

图的节点数组。

返回:

numpy.array,批处理图中图的累积节点和(第一个元素为0)。

property node_map_idx

节点的索引列表。

返回:

numpy.array,数组,指示每个节点的图索引。