fix.

apache · May 29, 2019 · 3e9893d · 3e9893d
1 parent c283601
commit 3e9893d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/mxnet/gluon/parameter.py b/python/mxnet/gluon/parameter.py
@@ -157,7 +157,9 @@ def grad_req(self, req):
 
     @property
     def shape(self):
-        if is_np_shape():
+        if self._shape is None:
+            return None
+        elif is_np_shape():
             return tuple(i if i != 0 else -1 for i in self._shape)
         else:
             return self._shape