Solved by albanD in post #4 
              
              
                Sorry I missed your conv2d as a static function, my bad. 
Few details: 

You nn.Module does not work for me and through the not implemented error from _conv_forward

You should never call the .forward() of the nn.Module directly. You should use the __call__ on it as  result = layer(image).
You don’t…