Moving eigendecomposition return from gpu to it's own line