gistfile1.py

class Conv_Module():
    def __init__(self, img, ker, b, pad=1, stride=1):            #initialization
        
        self.img_b, self.img_d, self.img_h, self.img_w=img.shape
        self.ker_b, self.ker_d, self.ker_h, self.ker_w=ker.shape
        self.b, self.pad, self.stride, self.img, self.ker=b, pad, stride, img, ker
        
    def __call__(self):                                      
        self.out=self.forward(self.img, self.ker)            #forward method is called with images, kernel as input
        return self.out
    
    def forward(self): raise Exception('Not Implemented')    #default raises not implemented error.
    def backward(self): return self.bwd(self.out)            #output from the forward method gets passed here as the input.