Add unit test for bilinear initializer

apache · Dec 18, 2018 · 5e59c11 · 5e59c11
1 parent 653f3e6
commit 5e59c11
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/python/unittest/test_init.py b/tests/python/unittest/test_init.py
@@ -60,8 +60,17 @@ def check_rsp_const_init(init, val):
     check_rsp_const_init(mx.initializer.Zero(), 0.)
     check_rsp_const_init(mx.initializer.One(), 1.)
 
+def test_bilinear_init():
+    bili = mx.init.Bilinear()
+    bili_weight = mx.ndarray.empty((1,1,4,4))
+    bili._init_weight(None, bili_weight)
+    bili_1d = np.array([[1/4, 3/4, 3/4, 1/4]])
+    bili_2d = bili_1d * np.transpose(bili_1d)
+    assert (np.abs(bili_2d - bili_weight.asnumpy()) < 1e-7).all()
+
 if __name__ == '__main__':
     test_variable_init()
     test_default_init()
     test_aux_init()
     test_rsp_const_init()
+    test_bilinear_init()