Khatri-Rao产品在Armadillo中正常使用

Question

我用C ++实现了Khatri-Rao产品或列式Kronecker产品。它运作良好，但是，我想知道是否有人知道如何使用Armadillo正确地进行操作。 谢谢您的时间：）

arma::mat kr_prod(arma::mat const &A, arma::mat const &B)
{
    arma::mat result = arma::mat(A.n_rows * B.n_rows, B.n_cols, arma::fill::zeros);
    for (int i = 0; i < A.n_cols; ++i)
    {
        result.col(i) = arma::kron(A.col(i), B.col(i));
    }
    return result;
}