#include "keras_model.h"

Inheritance diagram for keras::LayerFlatten:

Public Member Functions
	LayerFlatten ()

void	load_weights (std::ifstream &fin)

keras::DataChunk *	compute_output (keras::DataChunk *)

virtual unsigned int	get_input_rows () const

virtual unsigned int	get_input_cols () const

virtual unsigned int	get_output_units () const

std::string	get_name ()

Public Attributes
std::string	m_name

Detailed Description

Definition at line 136 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerFlatten::LayerFlatten ( )

inline

Definition at line 138 of file keras_model.h.

138 : Layer("Flatten") {}

keras::Layer::Layer

Layer(std::string name)

Definition: keras_model.h:124

Member Function Documentation

keras::DataChunk * keras::LayerFlatten::compute_output ( keras::DataChunk * dc )

virtual

Implements keras::Layer.

Definition at line 124 of file keras_model.cc.

References keras::DataChunkFlat::get_1d_rw(), and keras::DataChunk::get_3d().

                                                                     {
   vector<vector<vector<float> > > im = dc->get_3d();
 
   size_t csize = im[0].size();
   size_t rsize = im[0][0].size();
   size_t size = im.size() * csize * rsize;
   keras::DataChunkFlat *out = new DataChunkFlat(size);
   float * y_ret = out->get_1d_rw().data();
   for(size_t i = 0, dst = 0; i < im.size(); ++i) {
     for(size_t j = 0; j < csize; ++j) {
       float * row = im[i][j].data();
       for(size_t k = 0; k < rsize; ++k) {
         y_ret[dst++] = row[k];
       }
     }
   }
 
   return out;
 }