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


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


用法:

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

参数

  • graph(NDArray) - 我们对顶点进行采样的输入图。
  • data(NDArray[]) - 包含数据数组和状态的输入数组。
  • return_mapping(boolean, required) - 返回子图和父图之间的 vid 和 eid 映射。
  • out(NDArray, optional) - 输出 NDArray 来保存结果。

返回

out- 此函数的输出。

返回类型

NDArray 或 NDArray 列表

该运算符为图中的给定顶点集构造一个诱导子图。运算符接受多组顶点作为输入。对于每组顶点,如果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.ndarray.contrib.dgl_subgraph。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。