summaryrefslogtreecommitdiff
path: root/src/CuTNetLib/cuConcatenate.h
diff options
context:
space:
mode:
authorJoe Zhao <ztuowen@gmail.com>2014-04-14 08:14:45 +0800
committerJoe Zhao <ztuowen@gmail.com>2014-04-14 08:14:45 +0800
commitcccccbf6cca94a3eaf813b4468453160e91c332b (patch)
tree23418cb73a10ae3b0688681a7f0ba9b06424583e /src/CuTNetLib/cuConcatenate.h
downloadtnet-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.h90
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