I already tested my new layer with tensor operation in Python. I didnt’ have to implement the backward function since the layer is composed of basic tensor operations.
It works well, but too slow.
So, I want to boost the layer by writing it in CUDA.