From cccccbf6cca94a3eaf813b4468453160e91c332b Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Mon, 14 Apr 2014 08:14:45 +0800
Subject: First commit

---
 src/CuTNetLib/cuBlockArray.h | 90 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 src/CuTNetLib/cuBlockArray.h

(limited to 'src/CuTNetLib/cuBlockArray.h')

diff --git a/src/CuTNetLib/cuBlockArray.h b/src/CuTNetLib/cuBlockArray.h
new file mode 100644
index 0000000..5454538
--- /dev/null
+++ b/src/CuTNetLib/cuBlockArray.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
-- 
cgit v1.2.3-70-g09d2