Max pooling

pooling.py

def forward_pass(self, a_prev: np.array) -> np.array:
      self._shape = a_prev.shape
      n, h_in, w_in, c = a_prev.shape
      h_pool, w_pool = self._pool_size
      h_out = 1 + (h_in - h_pool) // self._stride
      w_out = 1 + (w_in - w_pool) // self._stride
      output = np.zeros((n, h_out, w_out, c))

      for i in range(h_out):
          for j in range(w_out):
              h_start = i * self._stride
              h_end = h_start + h_pool
              w_start = j * self._stride
              w_end = w_start + w_pool
              a_prev_slice = a_prev[:, h_start:h_end, w_start:w_end, :]
              self._save_mask(x=a_prev_slice, cords=(i, j))
              output[:, i, j, :] = np.max(a_prev_slice, axis=(1, 2))
      return output