Definition:
class etu { public: // build objects bool need_norm; int transfers_type = 2; int nbX; int nbY; int nbParams; arma::mat alpha; arma::mat gamma; arma::mat tau; arma::mat kappa; // kappa_ETU = -1/tau_inp arma::mat aux_alpha; // - alpha_ETU / tau_ETU = log(C_ETU) arma::mat aux_gamma; // - gamma_ETU / tau_ETU = log(D_ETU) arma::mat aux_alpha_exp; // exp(- alpha_ETU / tau_ETU), also labelled C_ETU arma::mat aux_gamma_exp; // exp(- gamma_ETU / tau_ETU), also labelled D_ETU // member functions void build(const arma::mat& alpha_ETU, const arma::mat& gamma_ETU, const arma::mat& tau_ETU, bool need_norm_ETU); void trans(); // // DSE-related functions arma::mat Psi(const arma::mat& U, const arma::mat& V) const; arma::mat Psi(const arma::mat& U, const arma::mat& V, arma::uvec* xs, arma::uvec* ys) const; arma::mat Psi(const double& U, const arma::mat& V, arma::uvec* xs, arma::uvec* ys) const; arma::mat Psi(const arma::mat& U, const double& V, arma::uvec* xs, arma::uvec* ys) const; double Psi(const double& U, const double& V, int x_ind, int y_ind) const; arma::mat du_Psi(const arma::mat& U, const arma::mat& V) const; arma::mat du_Psi(const arma::mat& U, const arma::mat& V, arma::uvec* xs, arma::uvec* ys) const; arma::mat du_Psi(const double& U, const arma::mat& V, arma::uvec* xs, arma::uvec* ys) const; arma::mat du_Psi(const arma::mat& U, const double& V, arma::uvec* xs, arma::uvec* ys) const; arma::mat dtheta_Psi(const arma::mat& U, const arma::mat& V, const arma::mat& dtheta); arma::mat dtheta_Psi(const arma::mat& U, const arma::mat& V, const arma::mat* dtheta); arma::mat Ucal(const arma::mat& vs) const; arma::mat Ucal(const arma::mat& vs, arma::uvec* xs, arma::uvec* ys) const; double Ucal(const double& vs, int xs, int ys) const; arma::mat Vcal(const arma::mat& us) const; arma::mat Vcal(const arma::mat& us, arma::uvec* xs, arma::uvec* ys) const; double Vcal(const double& us, int xs, int ys) const; arma::mat UW(const arma::mat& Ws) const; arma::mat UW(const arma::mat& Ws, arma::uvec* xs, arma::uvec* ys) const; double UW(const double& Ws, int x_ind, int y_ind) const; arma::mat VW(const arma::mat& Ws) const; arma::mat VW(const arma::mat& Ws, arma::uvec* xs, arma::uvec* ys) const; double VW(const double& Ws, int x_ind, int y_ind) const; arma::mat dw_UW(const arma::mat& Ws) const; arma::mat dw_UW(const arma::mat& Ws, arma::uvec* xs, arma::uvec* ys) const; arma::mat dw_VW(const arma::mat& Ws) const; arma::mat dw_VW(const arma::mat& Ws, arma::uvec* xs, arma::uvec* ys) const; arma::mat WU(const arma::mat& Us); arma::mat WU(const arma::mat& Us, arma::uvec* xs, arma::uvec* ys); arma::mat WV(const arma::mat& Vs); arma::mat WV(const arma::mat& Vs, arma::uvec* xs, arma::uvec* ys); };
Example: