add test.

tests/python/unittest/test_gluon.py

@@ -2726,6 +2726,23 @@ def hybrid_forward(self, F, x):
  net = Net(act0, act1, shape, slice)
  check_layer_forward_withinput(net, x)
 
+@with_seed()
+def test_np_shape_parameters():
+ class Foo(gluon.Block):
+ def __init__(self, **kwargs):
+ super(Foo, self).__init__(**kwargs)
+ self.dense = gluon.nn.Dense(16)
+ def forward(self, x):
+ return self.dense(x)
+
+ mx.set_np_compat(True)
+ z = mx.nd.zeros((2,2016))
+ print(z.shape)
+ foo = Foo()
+ foo.initialize()
+ print(foo(z).shape)
+ mx.set_np_compat(False)
+
 if __name__ == '__main__':
  import nose
  nose.runmodule()