/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

/* 
 * File:   RnnStack.hpp
 * Author: jens
 *
 * Created on 17. Januar 2022, 12:43
 */

#ifndef RNNSTACK_HPP
#define RNNSTACK_HPP

#include "AStack.hpp"
#include "matutils.hpp"

class RnnStack : public AStack
{
	const size_t NUM_CODES = 37;
	
public:
	RnnStack(const std::string &name);
	RnnStack(const RnnStack& orig) = delete;
	virtual ~RnnStack();

	size_t getSeqLen();

	void train(const arma::mat& batch, Rbm::IListener* pListener) override;
	
	arma::mat v_to_vc(const arma::mat &v, const arma::mat &c) const;
    arma::mat vc_to_v(const arma::mat &vc, size_t numContext) const;
	arma::mat step_forward(arma::mat &state, const arma::mat &v);

	void sample_one_hot(arma::mat &srcDst);
	void clamp_one_hot(arma::mat &srcDst);
	arma::mat to_next(const arma::mat &v);
	arma::mat to_curr(const arma::mat &v);

	void setParams(Rbm::Params const &param);

private:
};

#endif /* RNNSTACK_HPP */