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

@@ -1,9 +1,9 @@
 module Torch
   module NN
     class AvgPoolNd < Module
       def extra_inspect
-        format("kernel_size: %s", @kernel_size)
+        format("kernel_size: %s, stride: %s, padding: %s", @kernel_size, @stride, @padding)
       end
     end
   end
 end