Actually allocating tensor memory in extract_eigendecomposition