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: