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)