Update ch15_part1.py

for loop not following mathematical formula
oesmanhk · Apr 28, 2020 · cc3d11f · cc3d11f
1 parent 5d63c16
commit cc3d11f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ch15/ch15_part1.py b/ch15/ch15_part1.py
@@ -80,7 +80,7 @@ def conv1d(x, w, p=0, s=1):
         x_padded = np.concatenate(
             [zero_pad, x_padded, zero_pad])
     res = []
-    for i in range(0, int(len(x)/s),s):
+    for i in range(0, int((len(x_padded) - len(w_rot) / s)) + 1, s):
         res.append(np.sum(
             x_padded[i:i+w_rot.shape[0]] * w_rot))
     return np.array(res)