#include <Stack.h>

Collaboration diagram for lwtDev::ReductionStack:

Public Member Functions
	ReductionStack (size_t n_in, const std::vector< LayerConfig > &layers)
	~ReductionStack ()
	ReductionStack (ReductionStack &)=delete
ReductionStack &	operator= (ReductionStack &)=delete
VectorXd	reduce (MatrixXd inputs) const
size_t	n_outputs () const

Private Attributes
RecurrentStack *	m_recurrent
Stack *	m_stack

Detailed Description

Definition at line 195 of file Stack.h.

Constructor & Destructor Documentation

◆ ReductionStack() [1/2]

lwtDev::ReductionStack::ReductionStack	(	size_t	n_in,
		const std::vector< LayerConfig > &	layers )

Definition at line 396 of file Stack.cxx.

                                                                       {
    std::vector<LayerConfig> recurrent;
    std::vector<LayerConfig> feed_forward;
    std::set<Architecture> recurrent_arcs{
      Architecture::LSTM, Architecture::GRU, Architecture::EMBEDDING};
    for (const auto& layer: layers) {
      if (recurrent_arcs.count(layer.architecture)) {
        recurrent.push_back(layer);
      } else {
        feed_forward.push_back(layer);
      }
    }
    m_recurrent = new RecurrentStack(n_in, recurrent);
    m_stack = new Stack(m_recurrent->n_outputs(), feed_forward);
  }

◆ ~ReductionStack()

lwtDev::ReductionStack::~ReductionStack ( )

Definition at line 412 of file Stack.cxx.

                                  {
    delete m_recurrent;
    delete m_stack;
  }

◆ ReductionStack() [2/2]

lwtDev::ReductionStack::ReductionStack ( ReductionStack & )

delete

Member Function Documentation

◆ n_outputs()

size_t lwtDev::ReductionStack::n_outputs ( ) const

Definition at line 420 of file Stack.cxx.

                                         {
    return m_stack->n_outputs();
  }

◆ operator=()

ReductionStack & lwtDev::ReductionStack::operator= ( ReductionStack & )

delete

◆ reduce()

VectorXd lwtDev::ReductionStack::reduce ( MatrixXd inputs ) const

Definition at line 416 of file Stack.cxx.

                                                   {
    in = m_recurrent->scan(in);
    return m_stack->compute(in.col(in.cols() -1));
  }

Member Data Documentation

◆ m_recurrent

RecurrentStack* lwtDev::ReductionStack::m_recurrent

private

Definition at line 205 of file Stack.h.

◆ m_stack

Stack* lwtDev::ReductionStack::m_stack