/* * 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" class RnnStack : public AStack { public: RnnStack(const std::string &name); RnnStack(const std::string &name, Json::Value &layers, LayerConstructor *pLayerConstructor=nullptr); RnnStack(const RnnStack& orig) = delete; virtual ~RnnStack(); void train(const arma::mat& batch, Rbm::IListener* pListener) override; void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) override; private: }; #endif /* RNNSTACK_HPP */