Fixes #62636

kylematoba · kylematoba · commit bda17e751d7a · 2021-08-03T21:08:08.000+01:00
diff --git a/torch/csrc/api/include/torch/nn/options/pooling.h b/torch/csrc/api/include/torch/nn/options/pooling.h
@@ -29,7 +29,8 @@ struct AvgPoolOptions {
   /// when True, will include the zero-padding in the averaging calculation
   TORCH_ARG(bool, count_include_pad) = true;
 
-  /// if specified, it will be used as divisor, otherwise `kernel_size` will be used
+  /// if specified, it will be used as divisor, otherwise size of the pooling region will be used.
+
   TORCH_ARG(c10::optional<int64_t>, divisor_override) = c10::nullopt;
 };
 
diff --git a/torch/nn/modules/pooling.py b/torch/nn/modules/pooling.py
@@ -570,7 +570,8 @@ class AvgPool2d(_AvgPoolNd):
         padding: implicit zero padding to be added on both sides
         ceil_mode: when True, will use `ceil` instead of `floor` to compute the output shape
         count_include_pad: when True, will include the zero-padding in the averaging calculation
-        divisor_override: if specified, it will be used as divisor, otherwise :attr:`kernel_size` will be used
+        divisor_override: if specified, it will be used as divisor, otherwise size of the pooling region will be used.
+
 
     Shape:
         - Input: :math:`(N, C, H_{in}, W_{in})` or :math:`(C, H_{in}, W_{in})`.
