Could you provide a pytorch demo

[lll143653]

Hello Author @skokec , I am reading your paper and trying to understand the relevant codes, such as the translation of function in the paper, but the large number of cuda operations and some related codes based on caffe (caffe has not been maintained for a long time) make it difficult to read and connect with the formulas in the paper. Can you provide a simple pytorch demo?

best regards