lwtDev::MaxoutLayer Class Reference

#include <Stack.h>

Inheritance diagram for lwtDev::MaxoutLayer:

Collaboration diagram for lwtDev::MaxoutLayer:

Public Types
typedef std::pair< MatrixXd, VectorXd >	InitUnit

Public Member Functions
	MaxoutLayer (const std::vector< InitUnit > &maxout_tensor)
virtual VectorXd	compute (const VectorXd &) const override

Private Attributes
std::vector< MatrixXd >	m_matrices
MatrixXd	m_bias

Detailed Description

Definition at line 125 of file Stack.h.

Member Typedef Documentation

InitUnit

typedef std::pair<MatrixXd, VectorXd> lwtDev::MaxoutLayer::InitUnit

Definition at line 128 of file Stack.h.

Constructor & Destructor Documentation

MaxoutLayer()

lwtDev::MaxoutLayer::MaxoutLayer

(

const std::vector< InitUnit > &

maxout_tensor

)

Definition at line 217 of file Stack.cxx.

                                                                     :
    m_bias(units.size(), units.front().first.rows())
  {
    int out_pos = 0;
    for (const auto& unit: units) {
      m_matrices.push_back(unit.first);
      m_bias.row(out_pos) = unit.second;
      out_pos++;
    }
  }

Member Function Documentation

compute()

VectorXd lwtDev::MaxoutLayer::compute ( const VectorXd & in ) const

overridevirtual

Implements lwtDev::ILayer.

Definition at line 227 of file Stack.cxx.

                                                        {
    // eigen supports tensors, but only in the experimental component
    // for now just stick to matrix and vector classes
    const size_t n_mat = m_matrices.size();
    const size_t out_dim = m_matrices.front().rows();
    MatrixXd outputs(n_mat, out_dim);
    for (size_t mat_n = 0; mat_n < n_mat; mat_n++) {
      outputs.row(mat_n) = m_matrices.at(mat_n) * in;
    }
    outputs += m_bias;
    return outputs.colwise().maxCoeff();
  }

Member Data Documentation

m_bias

MatrixXd lwtDev::MaxoutLayer::m_bias

private

Definition at line 133 of file Stack.h.

m_matrices

std::vector<MatrixXd> lwtDev::MaxoutLayer::m_matrices

private

Definition at line 132 of file Stack.h.

---

The documentation for this class was generated from the following files:

• Stack.h
• Stack.cxx