C++ (Cpp) mat::eyeの例

コード例 #1

ファイル: nca_impl.hpp プロジェクト: DCooksey/SonicScalpel

void NCA<MetricType, OptimizerType>::LearnDistance(arma::mat& outputMatrix)
{
  // See if we were passed an initialized matrix.
  if ((outputMatrix.n_rows != dataset.n_rows) ||
      (outputMatrix.n_cols != dataset.n_rows))
    outputMatrix.eye(dataset.n_rows, dataset.n_rows);

  Timer::Start("nca_sgd_optimization");

  optimizer.Optimize(outputMatrix);

  Timer::Stop("nca_sgd_optimization");
}

コード例 #2

ファイル: main.cpp プロジェクト: DeepIntelligence/DeepLearning

 MyFunc2(int dim):ObjectFunc(dim){
     W.eye(dim,dim);
     b = arma::randu(dim);
     
     x_init = std::make_shared<arma::vec>(dim,arma::fill::randn);
 }