trả về nhiều ma trận trong c + + (thư viện Armadillo)

Question

Tôi đang làm việc với thư viện Armadillo trong C++. Lúc đầu, tôi tính toán một ma trận đặc biệt (trong mã của tôi: P), sau đó tôi tính toán phân tích QR (trong mã của tôi: Q). Cuối cùng tôi cần trả về cả P ​​và Q và một ma trận T khác cho hàm chính của tôi.

#include <iostream> 
#include <armadillo> 
using namespace std; 
using namespace arma; 
double phi(int n, int q){ 
... 
    mat P(n,n); 
    P=... 
    mat Q,R; 
    qr(Q,R,P); 
return P: 
return Q; 
return Q; 
... 
} 

int main() { 
    ... 
    int n,q; 
    cout<<"Enter the value of parameters n and q respectively:"<<endl; 
    cin>> n>>q; 
    phi(n,q); 
... 
} 

Tôi đang tìm kiếm một phương pháp để trả lại các ma trận trong armadillo với bằng cách sử dụng con trỏ và references.Here ma trận của tôi là rất lớn và thường là 500 * 500 hoặc 1000 * 1000. Có ai có giải pháp? Cảm ơn trước

Answer 1

Dưới đây là một ví dụ sử dụng std::tuple cùng với std::tie

#include <iostream> 
#include <armadillo> 

using namespace arma; 

std::tuple<mat, mat> phi(int const n, int const q) 
{ 
    ... 
    mat P(n, n); 
    P = ... 
    mat Q, R; 
    qr(Q, R, P); 
    return std::make_tuple(P, Q); 
} 

int main() 
{ 
    ... 
    int n, q; 
    std::cout << "Enter the value of parameters n and q respectively:" << std::endl; 
    std::cin >> n >> q; 
    mat P, Q; 
    std::tie(P, Q) = phi(n,q); 
    ... 
} 

Với C++ 17 (và một tuyên bố structured binding), bạn có thể làm điều đó như thế này:

#include <iostream> 
#include <armadillo> 

using namespace arma; 

std::tuple<mat, mat> phi(int const n, int const q) 
{ 
    ... 
    mat P(n, n); 
    P = ... 
    mat Q, R; 
    qr(Q, R, P); 
    return {P, Q}; 
} 

int main() 
{ 
    ... 
    int n,q; 
    std::cout << "Enter the value of parameters n and q respectively:" << std::endl; 
    std::cin >> n >> q; 
    auto [P, Q] = phi(n,q); 
    ... 
}