def forward(self, input): bw = binarize(self.weight) return F.conv2d(input, bw, self.bias, self.stride, self.padding, self.dilation, self.groups)