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

用法:
mxnet.contrib.ndarray.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.contrib.ndarray.dgl_subgraph。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：

示例：