当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.contrib.symbol.dgl_subgraph用法及代码示例


用法:

mxnet.contrib.symbol.dgl_subgraph(*data, **kwargs)

参数：

  • graph：(Symbol) - 我们对顶点进行采样的输入图。
  • data：(Symbol[]) - 包含数据数组和状态的输入数组。
  • return_mapping：(boolean, required) - 返回子图和父图之间的 vid 和 eid 映射。
  • name：(string, optional.) - 结果符号的名称。

返回：

结果符号。

返回类型：

Symbol

该运算符为图中的给定顶点集构造一个诱导子图。运算符接受多组顶点作为输入。对于每组顶点，如果return_mapping 为真，则返回一对 CSR 矩阵：第一个矩阵包含具有新边 ID 的边，第二个矩阵包含具有原始边 ID 的边。

示例：

x=[[1, 0, 0, 2],
  [3, 0, 4, 0],
  [0, 5, 0, 0],
  [0, 6, 7, 0]]
v = [0, 1, 2]
dgl_subgraph(x, v, return_mapping=True) =
  [[1, 0, 0],
   [2, 0, 3],
   [0, 4, 0]],
  [[1, 0, 0],
   [3, 0, 4],
   [0, 5, 0]]

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.contrib.symbol.dgl_subgraph。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。