c++ libtorch tensor 求解本征值和本征向量

原创已于 2024-10-22 16:56:09 修改 · 130 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

于 2024-10-22 16:55:34 首次发布

torch 专栏收录该内容

21 篇文章

订阅专栏

代码示例

torch::Tensor mat = torch::rand({ 2, 2, 4, 4}, torch::kComplexDouble); //2*2个4*4矩阵
std::tuple<torch::Tensor, torch::Tensor> eig_tuple = torch::linalg::eig(mat);
torch::Tensor eigenvalues = std::get<0>(eig_tuple); // 特征值
torch::Tensor eigenvectors = std::get<1>(eig_tuple); // 特征向量

torch::linalg::eig返回的是一个元组，其中第一个元素是包含本征值的Tensor，第二个元素是包含本征向量的Tensor。