If I want to implement some custom pooling or convolution layer, like the gated or tree pooling functions as shown in this article 64, do I have to modify the c++ code 38, instead of some simpler extension in python? I found the official website on ‘Custom C++ and CUDA Extensions’, but it seems not sufficient for my goal.