Set const value for output dim

apache · Feb 24, 2020 · 97f0523 · 97f0523
1 parent 0d0400e
commit 97f0523
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -38,6 +38,7 @@
 SMALL_X = 100
 SMALL_Y = 50
 LARGE_SIZE = LARGE_X * SMALL_Y
+LARGE_TENSOR_SHAPE = 4294967296
 
 
 def test_nn():
@@ -475,7 +476,7 @@ def check_embedding():
 
         out = nd.Embedding(data=data, weight=weight, input_dim=input_dim, output_dim=output_dim)
 
-        assert out.shape[0] == 4294967296
+        assert out.shape[0] == LARGE_TENSOR_SHAPE
         assert out.shape[1] == 1
         assert out.shape[2] == 1