lib/torch/nn/max_pool2d.rb in torch-rb-0.1.5 vs lib/torch/nn/max_pool2d.rb in torch-rb-0.1.6

@@ -1,9 +1,9 @@
 module Torch
   module NN
     class MaxPool2d < MaxPoolNd
       def forward(input)
-        F.max_pool2d(input, @kernel_size) # TODO other parameters
+        F.max_pool2d(input, @kernel_size, @stride, @padding, @dilation, @ceil_mode, @return_indices)
       end
     end
   end
 end