[Graph] small fix on the all index (#219)

python/dgl/graph.py

@@ -592,7 +592,9 @@ class DGLGraph(object):
         --------
         in_degree
         """
-        if not is_all(v):
+        if is_all(v):
+            v  = utils.toindex(slice(0, self.number_of_nodes()))
+        else:
             v = utils.toindex(v)
         return self._graph.in_degrees(v).tousertensor()
 
@@ -632,7 +634,9 @@ class DGLGraph(object):
         --------
         out_degree
         """
-        if not is_all(v):
+        if is_all(v):
+            v  = utils.toindex(slice(0, self.number_of_nodes()))
+        else:
             v = utils.toindex(v)
         return self._graph.out_degrees(v).tousertensor()
 

python/dgl/graph_index.py

@@ -395,9 +395,6 @@ class GraphIndex(object):
         int
             The in degree array.
         """
-        if is_all(v):
-            v = np.arange(0, self.number_of_nodes(), dtype=np.int64)
-            v = utils.toindex(v)
         v_array = v.todgltensor()
         return utils.toindex(_CAPI_DGLGraphInDegrees(self._handle, v_array))
 
@@ -429,9 +426,6 @@ class GraphIndex(object):
         int
             The out degree array.
         """
-        if is_all(v):
-            v = np.arange(0, self.number_of_nodes(), dtype=np.int64)
-            v = utils.toindex(v)
         v_array = v.todgltensor()
         return utils.toindex(_CAPI_DGLGraphOutDegrees(self._handle, v_array))
 

python/dgl/immutable_graph_index.py

@@ -365,7 +365,7 @@ class ImmutableGraphIndex(object):
             The in degree array.
         """
         deg = self._get_in_degree()
-        if is_all(v):
+        if v.is_slice(0, self.number_of_nodes()):
             return utils.toindex(deg)
         else:
             v_array = v.tousertensor()
@@ -401,7 +401,7 @@ class ImmutableGraphIndex(object):
             The out degree array.
         """
         deg = self._get_out_degree()
-        if is_all(v):
+        if v.is_slice(0, self.number_of_nodes()):
             return utils.toindex(deg)
         else:
             v_array = v.tousertensor()