C:/Users/albertlr/projects/TarzaNN/TarzaNN/GatingUnits.h

Go to the documentation of this file.

#ifndef GATINGUNITS_H_INCLUDED
#define GATINGUNITS_H_INCLUDED
#include <stdio.h>
#include <assert.h>
#include "MatrixOfMatrix.h"
#include "OutputImage.h"
#include "Filter.h"

class LearningFeaturePlane;

class GatingUnits:public MatrixOfMatrix<Matrix>{

public:
                
        GatingUnits( const int nXSize, const int nYSize, const int filterSizeX, const int filterSizeY, const int inputSizeX, const int inputSizeY);
        
    ~GatingUnits();
        
        float getGUMax();
        
        float getGUMin();
        
        OutputImage* stepGatingUnits(Matrix* gc, float max, float theta);
        OutputImage* stepGatingUnitsLearn(Matrix* gc, float max, float theta, LearningFeaturePlane* fp);

        inline void computeActivations(Matrix* activations){
                for(int gy=0; gy<height; gy++)
                        for(int gx=0; gx<width; gx++)
                                (*activations)(gx,gy)=(*this)(gx,gy)->sumAll();
        }

        void setZero();

};
typedef vector<GatingUnits*> GatingUnitsVector;
#endif /*GATINGUNITS_H_INCLUDED*/