Hi it seems that torch.eig method is deprecated. I have proposed the solution to the problem. So will try to update the code. Regards Piotr