lib/torch/nn/conv2d.rb in torch-rb-0.2.3 vs lib/torch/nn/conv2d.rb in torch-rb-0.2.4

- old
+ new

@@ -18,10 +18,17 @@ F.conv2d(input, @weight, @bias, @stride, @padding, @dilation, @groups) end # TODO add more parameters def extra_inspect - format("%s, %s, kernel_size: %s, stride: %s", @in_channels, @out_channels, @kernel_size, @stride) + s = String.new("%{in_channels}, %{out_channels}, kernel_size: %{kernel_size}, stride: %{stride}") + s += ", padding: %{padding}" if @padding != [0] * @padding.size + s += ", dilation: %{dilation}" if @dilation != [1] * @dilation.size + s += ", output_padding: %{output_padding}" if @output_padding != [0] * @output_padding.size + s += ", groups: %{groups}" if @groups != 1 + s += ", bias: false" unless @bias + s += ", padding_mode: %{padding_mode}" if @padding_mode != "zeros" + format(s, **dict) end end end end