COSMIT slightly less confusing repr for CplxParameter

hjbiao09 · Dec 21, 2019 · 317aeaf · 317aeaf
1 parent 982b146
commit 317aeaf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cplxmodule/layers.py b/cplxmodule/layers.py
@@ -33,6 +33,9 @@ def __init__(self, cplx):
         # save reference to the underlying cplx data
         self._cplx = cplx
 
+    def extra_repr(self):
+        return repr(tuple(self._cplx.shape))[1:-1]
+
     @property
     def data(self):
         return self._cplx