adding atan2

Author: ehsanhaghighat

SciANN.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: SciANN
-Version: 0.6.8.5
+Version: 0.6.8.6
 Summary: A Keras/Tensorflow wrapper for scientific computations and physics-informed deep learning using artificial neural networks.
 Home-page: https://github.com/sciann/sciann
 Author: Ehsan Haghighat

sciann/__init__.py

@@ -41,7 +41,7 @@
 __credits__ = []
 __url__ = "http://github.com/sciann/sciann]"
 __license__ = "MIT"
-__version__ = "0.6.8.5"
+__version__ = "0.6.8.6"
 __cite__ = \
     '@article{haghighat2021sciann, \n' +\
     '    title={SciANN: A Keras/TensorFlow wrapper for scientific computations and physics-informed deep learning using artificial neural networks}, \n' +\

sciann/utils/math.py

@@ -602,16 +602,37 @@ def atan(x):
     return _apply_function(x, 'atan')
 
 
-def atan2(x):
-    """Computes atan2 of x element-wise.
+def atan2(y, x):
+    """Computes atan2 of y, x pair, element-wise.
 
     # Arguments
+        y: Functional object.
         x: Functional object.
 
     # Returns
         A new functional object.
     """
-    return _apply_function(x, 'atan2')
+    validate_functional(x)
+    validate_functional(y)
+
+    fun = get_activation('atan2')
+    lmbd, outputs = [], []
+    for i in range(len(x.outputs)):
+        lmbd.append(
+            Lambda(
+                lambda xs: fun(xs[0], xs[1]),
+                name=graph_unique_name("{}".format('atan2'))
+            )
+        )
+        outputs += [lmbd[-1]([yi, xi]) for yi, xi in zip(y.outputs, x.outputs)]
+    Functional = x.get_class()
+    inputs = y.inputs.copy() + x.inputs.copy()
+    res = Functional(
+        inputs = unique_tensors(inputs),
+        outputs = outputs,
+        layers = lmbd
+    )
+    return res
 
 
 def cot(x):

setup.py

@@ -15,7 +15,7 @@
 
 setup(
     name='SciANN',
-    version='0.6.8.5',
+    version='0.6.8.6',
     description='A Keras/Tensorflow wrapper for scientific computations and physics-informed deep learning using artificial neural networks.',
     long_description=long_description,
     author='Ehsan Haghighat',