MaxPool2d.patch 645 B

1234567891011121314151617
  1. --- /usr/local/lib/python3.5/dist-packages/torch/nn/modules/pooling.py
  2. +++ /usr/local/lib/python3.5/dist-packages/torch/nn/modules/pooling.py
  3. @@ -57,12 +57,8 @@
  4. https://github.com/vdumoulin/conv_arithmetic/blob/master/README.md
  5. """
  6. - kernel_size: _size_2_t
  7. - stride: _size_2_t
  8. - padding: _size_2_t
  9. - dilation: _size_2_t
  10. -
  11. - def forward(self, input: Tensor) -> Tensor:
  12. + @weak_script_method
  13. + def forward(self, input):
  14. return F.max_pool2d(input, self.kernel_size, self.stride,
  15. self.padding, self.dilation, self.ceil_mode,
  16. self.return_indices)