utils.h

Go to the documentation of this file.

 
#pragma once
#include "../Matrix.h"
#include "../sorting.h"
#include <vector>
 
#ifndef EPS
  #define EPS 1e-8
#endif
 
 
Matrix<double> linspace(double start, double end, unsigned long num_elements);
 
Matrix<double> zeros(size_t rows, size_t columns, size_t elements = 1);
 
Matrix<double> zerosV(size_t rows);
 
Matrix<double> ones(size_t rows, size_t columns = 1, size_t elements = 1);
 
Matrix<double> eye(size_t rows, size_t columns = 0);
 
Matrix<double> tridiag(size_t rows, size_t columns, double left, double center, double right);
 
double norm(const Matrix<double>& in);
 
Matrix<double> norm(const Matrix<double>& in, int axis);
 
Matrix<size_t> argsort(const Matrix<double>& in);
 
 
Matrix<size_t> nonzero(const std::function<bool(const double&)>& validation, const Matrix<double>& x);
 
 
template<typename T>
Matrix<T> diag(const Matrix<T>& in) {
  bool isRows = in.rows() > in.columns();
  auto k      = isRows ? in.rows() : in.columns();
  auto result = eye(k);
  for(size_t i = 0; i < k; ++i) { result(i, i) = in(isRows ? i : 0, isRows ? 0 : i); }
  return result;
}