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 144 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerFlatten::LayerFlatten ( )

inline

Definition at line 146 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 128 of file keras_model.cc.

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

{
  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 154 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 150 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 140 of file keras_model.h.

{ return m_name; }

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

inlinevirtual

Implements keras::Layer.

Definition at line 155 of file keras_model.h.

{ return 0; }

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

inlinevirtual

Implements keras::Layer.

Definition at line 147 of file keras_model.h.

{};

Member Data Documentation

std::string keras::Layer::m_name

inherited

Definition at line 141 of file keras_model.h.

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.h
• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.cc