lib/torch/nn/conv2d.rb in torch-rb-0.1.4 vs lib/torch/nn/conv2d.rb in torch-rb-0.1.5

- old
+ new

@@ -1,9 +1,7 @@ module Torch module NN class Conv2d < ConvNd - attr_reader :bias, :weight - def initialize(in_channels, out_channels, kernel_size, stride: 1, padding: 0, dilation: 1, groups: 1, bias: true, padding_mode: "zeros") kernel_size = pair(kernel_size) stride = pair(stride) padding = pair(padding) dilation = pair(dilation)