diff options
author | Joe Zhao <ztuowen@gmail.com> | 2014-04-14 08:14:45 +0800 |
---|---|---|
committer | Joe Zhao <ztuowen@gmail.com> | 2014-04-14 08:14:45 +0800 |
commit | cccccbf6cca94a3eaf813b4468453160e91c332b (patch) | |
tree | 23418cb73a10ae3b0688681a7f0ba9b06424583e /src/CuTNetLib/cuConcatenate.h | |
download | tnet-cccccbf6cca94a3eaf813b4468453160e91c332b.tar.gz tnet-cccccbf6cca94a3eaf813b4468453160e91c332b.tar.bz2 tnet-cccccbf6cca94a3eaf813b4468453160e91c332b.zip |
First commit
Diffstat (limited to 'src/CuTNetLib/cuConcatenate.h')
-rw-r--r-- | src/CuTNetLib/cuConcatenate.h | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/src/CuTNetLib/cuConcatenate.h b/src/CuTNetLib/cuConcatenate.h new file mode 100644 index 0000000..5454538 --- /dev/null +++ b/src/CuTNetLib/cuConcatenate.h @@ -0,0 +1,90 @@ +#ifndef _CUBLOCK_ARRAY_H_ +#define _CUBLOCK_ARRAY_H_ + + +#include "cuComponent.h" +#include "cumatrix.h" + +#include "Matrix.h" +#include "Vector.h" + + +namespace TNet { + + class CuNetwork; + /** + * \brief Updatable compenent consists of several networks + * + * \ingroup CuNNUpdatable + * Each network is individually propagated and backpropagated with non-overlaping input and output. + * + * Enabling multipath topological structure within the network. + */ + class CuBlockArray : public CuUpdatableComponent + { + public: + + CuBlockArray(size_t nInputs, size_t nOutputs, CuComponent *pPred); + ~CuBlockArray(); + + ComponentType GetType() const; + const char* GetName() const; + + void PropagateFnc(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y); + void BackpropagateFnc(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y); + + void Update(); + + void ReadFromStream(std::istream& rIn); + void WriteToStream(std::ostream& rOut); + + protected: + std::vector<CuNetwork*> mBlocks; ///< vector with networks, one network is one block + size_t mNBlocks; + }; + + + + + //////////////////////////////////////////////////////////////////////////// + // INLINE FUNCTIONS + // CuBlockArray:: + inline + CuBlockArray:: + CuBlockArray(size_t nInputs, size_t nOutputs, CuComponent *pPred) + : CuUpdatableComponent(nInputs, nOutputs, pPred), + mNBlocks(0) + { } + + + inline + CuBlockArray:: + ~CuBlockArray() + { + for(int i=0; i<mBlocks.size(); i++) { + delete mBlocks[i]; + } + mBlocks.clear(); + } + + inline CuComponent::ComponentType + CuBlockArray:: + GetType() const + { + return CuComponent::BLOCK_ARRAY; + } + + inline const char* + CuBlockArray:: + GetName() const + { + return "<blockarray>"; + } + + + +} //namespace + + + +#endif |