Solved by albanD in post #3 
              
              
                Hi, 
I am afraid this is not something that can easily be backported to older versions of pytorch as they don’t have complex support. 
You will have to use 1.6+ I’m afraid