keras::LayerFlatten Class Reference

#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.

: Layer("Flatten") {}

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;
}

virtual unsigned int keras::LayerFlatten::get_input_cols ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 143 of file keras_model.h.

{ return 0; } // same as for rows

virtual unsigned int keras::LayerFlatten::get_input_rows ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 142 of file keras_model.h.

{ return 0; } // look for the value in the preceding layer

std::string keras::Layer::get_name ( )

inlineinherited

Definition at line 131 of file keras_model.h.

{ return m_name; }

virtual unsigned int keras::LayerFlatten::get_output_units ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 144 of file keras_model.h.

{ return 0; }

void keras::LayerFlatten::load_weights ( std::ifstream &  fin )

inlinevirtual

Implements keras::Layer.

Definition at line 139 of file keras_model.h.

{};

Member Data Documentation

std::string keras::Layer::m_name

inherited

Definition at line 132 of file keras_model.h.

---

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

• /cvmfs/larsoft.opensciencegrid.org/products/larreco/v07_10_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.h
• /cvmfs/larsoft.opensciencegrid.org/products/larreco/v07_10_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.cc