#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