[Transform] Modules for Augmentation (dmlc#3668)

* Update * Update * Fix * Update * Update * Update * Update * Fix * Update * Update * Update * Update * Fix lint * lint * Update * Update * lint fix * Fix CI * Fix * Fix CI * Update * Fix * Update * Update * Augmentation (dmlc#10) * Update * PPR * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * CI * lint * lint * Update * Update * Fix AddEdge * try import * Update * Fix * CI Co-authored-by: Ubuntu <[email protected]> Co-authored-by: Minjie Wang <[email protected]>
sanchit-misra · Jan 25, 2022 · 2b98e76 · 2b98e76
1 parent ba62b73
commit 2b98e76
Show file tree

Hide file tree

Showing 8 changed files with 830 additions and 8 deletions.
diff --git a/docs/source/api/python/transform.rst b/docs/source/api/python/transform.rst
@@ -13,6 +13,12 @@ BaseTransform
     :members: __call__, __repr__
     :show-inheritance:
 
+Compose
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: Compose
+    :show-inheritance:
+
 AddSelfLoop
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
@@ -55,8 +61,50 @@ AddMetaPaths
 .. autoclass:: AddMetaPaths
     :show-inheritance:
 
-KNNGraph
+GCNNorm
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: GCNNorm
+    :show-inheritance:
+
+PPR
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: PPR
+    :show-inheritance:
+
+HeatKernel
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: HeatKernel
+    :show-inheritance:
+
+GDC
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: GDC
+    :show-inheritance:
+
+NodeShuffle
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: NodeShuffle
+    :show-inheritance:
+
+DropNode
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: DropNode
+    :show-inheritance:
+
+DropEdge
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: DropEdge
+    :show-inheritance:
+
+AddEdge
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
-.. autoclass:: KNNGraph
+.. autoclass:: AddEdge
     :show-inheritance:
diff --git a/python/dgl/backend/backend.py b/python/dgl/backend/backend.py
@@ -570,6 +570,21 @@ def exp(input):
     """
     pass
 
+def inverse(input):
+    """Returns the inverse matrix of a square matrix if it exists.
+
+    Parameters
+    ----------
+    input : Tensor
+        The input square matrix.
+
+    Returns
+    -------
+    Tensor
+        The output tensor.
+    """
+    pass
+
 def sqrt(input):
     """Returns a new tensor with the square root of the elements of the input tensor `input`.
 
@@ -1057,6 +1072,21 @@ def equal(x, y):
     """
     pass
 
+def allclose(x, y, rtol=1e-4, atol=1e-4):
+    """Compares whether all elements are close.
+
+    Parameters
+    ----------
+    x : Tensor
+        First tensor
+    y : Tensor
+        Second tensor
+    rtol : float, optional
+        Relative tolerance
+    atol : float, optional
+        Absolute tolerance
+    """
+
 def logical_not(input):
     """Perform a logical not operation.  Equivalent to np.logical_not
 

diff --git a/python/dgl/backend/mxnet/tensor.py b/python/dgl/backend/mxnet/tensor.py
@@ -191,6 +191,9 @@ def argsort(input, dim, descending):
 def exp(input):
     return nd.exp(input)
 
+def inverse(input):
+    return nd.linalg_inverse(input)
+
 def sqrt(input):
     return nd.sqrt(input)
 
@@ -327,6 +330,9 @@ def boolean_mask(input, mask):
 def equal(x, y):
     return x == y
 
+def allclose(x, y, rtol=1e-4, atol=1e-4):
+    return np.allclose(x.asnumpy(), y.asnumpy(), rtol=rtol, atol=atol)
+
 def logical_not(input):
     return nd.logical_not(input)
 

diff --git a/python/dgl/backend/pytorch/tensor.py b/python/dgl/backend/pytorch/tensor.py
@@ -14,8 +14,8 @@
 from ...function.base import TargetCode
 from ...base import dgl_warning
 
-if LooseVersion(th.__version__) < LooseVersion("1.5.0"):
-    raise Exception("Detected an old version of PyTorch. Please update torch>=1.5.0 "
+if LooseVersion(th.__version__) < LooseVersion("1.8.0"):
+    raise Exception("Detected an old version of PyTorch. Please update torch>=1.8.0 "
                     "for the best experience.")
 
 def data_type_dict():
@@ -164,6 +164,9 @@ def argtopk(input, k, dim, descending=True):
 def exp(input):
     return th.exp(input)
 
+def inverse(input):
+    return th.inverse(input)
+
 def sqrt(input):
     return th.sqrt(input)
 
@@ -276,6 +279,9 @@ def boolean_mask(input, mask):
 def equal(x, y):
     return x == y
 
+def allclose(x, y, rtol=1e-4, atol=1e-4):
+    return th.allclose(x, y, rtol=rtol, atol=atol)
+
 def logical_not(input):
     return ~input
 

diff --git a/python/dgl/backend/tensorflow/tensor.py b/python/dgl/backend/tensorflow/tensor.py
@@ -244,6 +244,10 @@ def exp(input):
     return tf.exp(input)
 
 
+def inverse(input):
+    return tf.linalg.inv(input)
+
+
 def sqrt(input):
     return tf.sqrt(input)
 
@@ -396,6 +400,11 @@ def equal(x, y):
     return x == y
 
 
+def allclose(x, y, rtol=1e-4, atol=1e-4):
+    return np.allclose(tf.convert_to_tensor(x).numpy(),
+                       tf.convert_to_tensor(y).numpy(), rtol=rtol, atol=atol)
+
+
 def logical_not(input):
     return ~input