Implement functions to create our desired matrix

This commit is contained in:
2023-09-16 18:01:26 +02:00
parent 97d13c6d2d
commit 68e665748a
2 changed files with 54 additions and 0 deletions

16
include/matrix.hpp Normal file
View File

@@ -0,0 +1,16 @@
#ifndef __MATRIX__
#define __MATRIX__
#include <armadillo>
arma::mat create_tridiagonal(
const arma::vec& a,
const arma::vec& d,
const arma::vec& e);
arma::mat create_tridiagonal(int n, double a, double d, double e);
arma::mat create_symmetric_tridiagonal(int n, double a, double d);
#endif