complete DGMR code

[TQRTQ]

The code is great, I can understand the details of the paper better, but I found a lot of code missing in layer.txt, such as SNConv3d, ApplyAlongAxis. I have finished a Pytorch version of DGMR code and uploaded it to my GitHub, https://github.com/TQRTQ/DGMR after looking at the pseudo code I found some details that I misunderstood. I want to reproduce this work, can you give me the complete code？please！