lib/torch/nn/max_poolnd.rb in torch-rb-0.2.3 vs lib/torch/nn/max_poolnd.rb in torch-rb-0.2.4
- old
+ new
@@ -10,10 +10,11 @@
@return_indices = return_indices
@ceil_mode = ceil_mode
end
def extra_inspect
- format("kernel_size: %s", @kernel_size)
+ s = "kernel_size: %{kernel_size}, stride: %{stride}, padding: %{padding}, dilation: %{dilation}, ceil_mode: %{ceil_mode}"
+ format(s, **dict)
end
end
end
end