include/matrixlinker.h

Go to the documentation of this file.

/********************************************************************************
 *  Neural Network Framework.                                                   *
 *  Copyright (C) 2005-2008 Gianluca Massera <emmegian@yahoo.it>                *
 *                                                                              *
 *  This program is free software; you can redistribute it and/or modify        *
 *  it under the terms of the GNU General Public License as published by        *
 *  the Free Software Foundation; either version 2 of the License, or           *
 *  (at your option) any later version.                                         *
 *                                                                              *
 *  This program is distributed in the hope that it will be useful,             *
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of              *
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               *
 *  GNU General Public License for more details.                                *
 *                                                                              *
 *  You should have received a copy of the GNU General Public License           *
 *  along with this program; if not, write to the Free Software                 *
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA  *
 ********************************************************************************/

#ifndef MATRIXLINKER_H
#define MATRIXLINKER_H

#include "types.h"
#include "linker.h"

namespace nnfw {

class NNFW_API MatrixLinker : public Linker {
public:

    MatrixLinker( Cluster* from, Cluster* to, const char* name = "unnamed" );

    MatrixLinker( PropertySettings& prop );

    virtual ~MatrixLinker();



    u_int rows() {
        return nrows;
    };

    u_int cols() {
        return ncols;
    };

    u_int size() const;

    virtual void randomize( Real min, Real max );

    virtual void setWeight( u_int from, u_int to, Real weight );

    virtual Real getWeight( u_int from, u_int to );

    RealMat& matrix() {
        return w;
    };

    const RealMat& matrix() const {
        return w;
    };

    Variant matrixP() {
        return Variant( &w );
    };

    void setMatrix( const RealMat& mat );

    bool setMatrix( const Variant& v );


private:
    u_int nrows, ncols;
    RealMat w;
};

}

#endif

Copyright © 2005-2008 G. Massera

Contact: emmegian (at) yahoo (dot) it

  Documentation generated by