custom.py 文件源码

def forward(self, input, kernel):
        self.weight = Parameter(kernel.data)
        # print 'weight: ', self.weight.size()
        # print 'bias: ', self.bias.size()
        # print 'forward:', type(input.data), type(self.weight.data)
        # print 'forward: ', input.size(), self.weight.size()
        return F.conv2d(input, kernel, self.bias, self.stride,
                        self.padding, self.dilation, self.groups)