Add stack for tensorflow.

lululxvi · Jun 12, 2023 · 43d755c · 43d755c
1 parent 85abdef
commit 43d755c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepxde/backend/tensorflow/tensor.py b/deepxde/backend/tensorflow/tensor.py
@@ -75,6 +75,10 @@ def to_numpy(input_tensor):
     return input_tensor.numpy()
 
 
+def stack(values, axis):
+    return tf.stack(values, axis)
+
+
 def elu(x):
     return tf.nn.elu(x)