Thanks for the useful information!
A follow-up question is: If I want to write a modified version of col2im() with a different name, can I write it in C++ extensions? Or must I write it inside the ATen library and recompile?