Re-organize GeMOSA

2021-05-27 15:56:08 +08:00
parent 6da60664f5
commit 4ca38ad2ff
6 changed files with 26 additions and 19 deletions
--- a/xautodl/datasets/math_base_funcs.py
+++ b/xautodl/datasets/math_base_funcs.py
@@ -40,6 +40,9 @@ class MathFunc(abc.ABC):
    def reset_xstr(self, xstr):
        self._xstr = str(xstr)

+    def output_shape(self, input_shape):
+        return input_shape
+
    @abc.abstractmethod
    def __call__(self, x):
        raise NotImplementedError