fix dtype and shape issue in avg_pool1d numpy backend

IgorTatarnikov · Apr 5, 2023 · dafba54 · dafba54
1 parent 5d743bf
commit dafba54
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ivy/functional/backends/numpy/experimental/layers.py b/ivy/functional/backends/numpy/experimental/layers.py
@@ -266,7 +266,7 @@ def avg_pool1d(
             (pad_w // 2, pad_w - pad_w // 2),
             (0, 0),
         ],
-        "edge",
+        constant_values=0.0,
     )
 
     x_shape = x.shape
@@ -289,15 +289,17 @@ def avg_pool1d(
         _get_num_padded_values,
         constant={
             "p": pad_w,
-            "n": x_shape[1],
+            "n": x.shape[1] - pad_w,
             "k": kernel[0],
             "s": strides[0],
         },
         unique={
             "i": np.arange(res.shape[1]),
         },
     )
-    res = kernel[0] * res / (kernel[0] - num_padded_values)
+    res = (kernel[0] * res) / (
+        kernel[0] - np.array(num_padded_values, dtype=res.dtype)
+    )[:, None]
 
     if data_format == "NCW":
         return res.swapaxes(1, 2)