fix: normalize flag wasn't documented (#663)

Co-authored-by: dante <[email protected]>

src/tensor/ops.rs

@@ -2262,6 +2262,7 @@ pub fn deconv<
 /// * `padding` - Tuple of padding values in x and y directions.
 /// * `stride` - Tuple of stride values in x and y directions.
 /// * `pool_dims` - Tuple of pooling window size in x and y directions.
+/// * `normalize` - Flag to normalize the output by the number of elements in the pooling window.
 /// # Examples
 /// ```
 /// use ezkl::tensor::Tensor;
@@ -2277,6 +2278,11 @@ pub fn deconv<
 /// let pooled = sumpool(&x, [(0, 0); 2], (1, 1), (2, 2), false).unwrap().0;
 /// let expected: Tensor<i128> = Tensor::<i128>::new(Some(&[11, 8, 8, 10]), &[1, 1, 2, 2]).unwrap();
 /// assert_eq!(pooled, expected);
+///
+/// // This time with normalization
+/// let pooled = sumpool(&x, [(0, 0); 2], (1, 1), (2, 2), true).unwrap().0;
+/// let expected: Tensor<i128> = Tensor::<i128>::new(Some(&[3, 2, 2, 3]), &[1, 1, 2, 2]).unwrap();
+/// assert_eq!(pooled, expected);
 /// ```
 pub fn sumpool(
     image: &Tensor<i128>,