Есть ли способ сортировки матрицы на основе другой матрицы или вектора в pytorch? Учитывая собственные значения и собственные векторы, как показано ниже:
eigenvalues, eigenvectors = torch.eig(matrix,eigenvectors=True) #compute the eigenvalues & vectors
sortedEig, indices=torch.sort(w, dim=1, descending=True)
eigenvectors_sorted = lamb.gather(dim=1, index=indices)
Мне нужно отсортировать собственные векторы по убыванию собственных значений. Код выше показывает, что я попробовал. Однако вывод eigenvectors_sorted представляет собой N x 2 тензор вместо N x N тензор , который на самом деле собственных векторов .
Обратите внимание, что в pytorch собственные значения представляют собой N x 2 тензор .