Sample python bilinear initializer at integral points in y-direction

apache · Dec 11, 2018 · 653f3e6 · 653f3e6
1 parent ee5f699
commit 653f3e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/initializer.py b/python/mxnet/initializer.py
@@ -216,7 +216,7 @@ def _init_bilinear(self, _, arr):
         c = (2 * f - 1 - f % 2) / (2. * f)
         for i in range(np.prod(shape)):
             x = i % shape[3]
-            y = (i / shape[3]) % shape[2]
+            y = (i // shape[3]) % shape[2]
             weight[i] = (1 - abs(x / f - c)) * (1 - abs(y / f - c))
         arr[:] = weight.reshape(shape)
 
@@ -656,7 +656,7 @@ def _init_weight(self, _, arr):
         c = (2 * f - 1 - f % 2) / (2. * f)
         for i in range(np.prod(shape)):
             x = i % shape[3]
-            y = (i / shape[3]) % shape[2]
+            y = (i // shape[3]) % shape[2]
             weight[i] = (1 - abs(x / f - c)) * (1 - abs(y / f - c))
         arr[:] = weight.reshape(shape)