Hi Marco!

As far as I understand the author have done the same thing(I am taking about their longer code implementation in the last page of the paper), the only difference is that they used it in a nn.Sequential block and they run it "depth" times, so aren't they reusing them?

Correct me If I am wrong and thanks for reviewing the code