ntf_utils.hpp

Go to the documentation of this file.

/* Copyright 2017 Ramakrishnan Kannan */

#ifndef COMMON_NTF_UTILS_HPP_
#define COMMON_NTF_UTILS_HPP_

#include "common/ncpfactors.hpp"
#include "common/tensor.hpp"

void khatrirao(const MAT &i_A, const MAT &i_B, MAT *o_C) {
  assert(i_A.n_cols == i_B.n_cols);
  VEC acol = arma::zeros<VEC>(i_A.n_rows);
  VEC bcol = arma::zeros<VEC>(i_B.n_rows);
  for (unsigned int i = 0; i < i_A.n_cols; i++) {
    acol = i_A.col(i);
    bcol = i_B.col(i);
    for (unsigned int j = 0; j < acol.n_rows; j++) {
      (*o_C)(arma::span(j * bcol.n_rows, (j + 1) * bcol.n_rows - 1), i) =
          acol(j) * bcol;
    }
  }
}
inline void kronecker(const VEC &i_acol, const VEC &i_bcol, VEC *o_c) {
  for (unsigned int j = 0; j < i_acol.n_rows; j++) {
    (*o_c).rows(arma::span(j * i_bcol.n_rows, (j + 1) * i_bcol.n_rows - 1)) =
        i_acol(j) * i_bcol;
  }
}
void mttkrp(const int i_n, const planc::Tensor &X, planc::NCPFactors &i_F,
            MAT *o_mttkrp) {
  MAT krp = i_F.krp_leave_out_one(i_n);
  X.mttkrp(i_n, krp, o_mttkrp);
}

#endif // COMMON_NTF_UTILS_HPP_