Commit dafe4671 authored by Da Zheng's avatar Da Zheng Committed by Minjie Wang
Browse files

[Bug][MXNet] fix an API in immutable graph index. (#233)

parent b1e8d95e
...@@ -479,18 +479,37 @@ class ImmutableGraphIndex(object): ...@@ -479,18 +479,37 @@ class ImmutableGraphIndex(object):
return F.copy_to(new_mat, ctx) return F.copy_to(new_mat, ctx)
return self._sparse.adjacency_matrix(transpose, ctx) return self._sparse.adjacency_matrix(transpose, ctx)
def incidence_matrix(self, oriented=False): def incidence_matrix(self, type, ctx):
"""Return the incidence matrix representation of this graph. """Return the incidence matrix representation of this graph.
An incidence matrix is an n x m sparse matrix, where n is
the number of nodes and m is the number of edges. Each nnz
value indicating whether the edge is incident to the node
or not.
There are three types of an incidence matrix `I`:
* "in":
- I[v, e] = 1 if e is the in-edge of v (or v is the dst node of e);
- I[v, e] = 0 otherwise.
* "out":
- I[v, e] = 1 if e is the out-edge of v (or v is the src node of e);
- I[v, e] = 0 otherwise.
* "both":
- I[v, e] = 1 if e is the in-edge of v;
- I[v, e] = -1 if e is the out-edge of v;
- I[v, e] = 0 otherwise (including self-loop).
Parameters Parameters
---------- ----------
oriented : bool, optional (default=False) type : str
Whether the returned incidence matrix is oriented. Can be either "in", "out" or "both"
ctx : context
The context of returned incidence matrix.
Returns Returns
------- -------
utils.CtxCachedObject SparseTensor
An object that returns tensor given context. The incidence matrix.
""" """
raise Exception('immutable graph doesn\'t support incidence_matrix for now.') raise Exception('immutable graph doesn\'t support incidence_matrix for now.')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment