00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
#include "ConcatColumnsVariable.h"
00044
#include "SubMatVariable.h"
00045
00046
namespace PLearn {
00047
using namespace std;
00048
00049
00052
PLEARN_IMPLEMENT_OBJECT(ConcatColumnsVariable,
00053
"Concatenation of the columns of several variables",
00054
"NO HELP");
00055
00056 ConcatColumnsVariable::ConcatColumnsVariable(
const VarArray& vararray)
00057 :
inherited(vararray.nonNull(), vararray.maxLength(), vararray.sumOfWidths())
00058 {
00059
build_();
00060 }
00061
00062
void
00063 ConcatColumnsVariable::build()
00064 {
00065 inherited::build();
00066
build_();
00067 }
00068
00069
void
00070 ConcatColumnsVariable::build_()
00071 {
00072
if (varray->
length()) {
00073
int l = varray[0]->
length();
00074
for (
int i = 1; i < varray.size(); i++)
00075
if (l != varray[i]->length())
00076
PLERROR(
"ConcatColumnsVariable: all non-null variables must have the same length");
00077 }
00078 }
00079
00080 void ConcatColumnsVariable::recomputeSize(
int& l,
int& w)
const
00081
{
00082
if (varray) {
00083 l = varray.
maxLength();
00084 w = varray.
sumOfWidths();
00085 }
else
00086 l = w = 0;
00087 }
00088
00089 void ConcatColumnsVariable::fprop()
00090 {
00091
int n_rows = matValue.
length();
00092
int m_start = 0;
00093
int mod = matValue.
mod();
00094
for (
int m=0;m<varray.
size();m++)
00095 {
00096
real* mp = varray[m]->valuedata;
00097
int n_cols = varray[m]->matValue.width();
00098
real* p = &valuedata[m_start];
00099
for (
int i=0;i<n_rows;i++,p+=mod)
00100
for (
int j=0;j<n_cols;j++,mp++)
00101 p[j] = *mp;
00102 m_start+=n_cols;
00103 }
00104 }
00105
00106
00107 void ConcatColumnsVariable::bprop()
00108 {
00109
int n_rows = matValue.
length();
00110
int m_start = 0;
00111
int mod = matValue.
mod();
00112
for (
int m=0;m<varray.
size();m++)
00113 {
00114
real* mp = varray[m]->gradientdata;
00115
int n_cols = varray[m]->matGradient.width();
00116
real* p = &gradientdata[m_start];
00117
for (
int i=0;i<n_rows;i++,p+=mod)
00118
for (
int j=0;j<n_cols;j++,mp++)
00119 *mp += p[j];
00120 m_start+=n_cols;
00121 }
00122 }
00123
00124
00125 void ConcatColumnsVariable::symbolicBprop()
00126 {
00127
int k=0;
00128
for (
int n=0; n<varray.
size(); n++) {
00129
Var vn = varray[n];
00130 vn->accg(
new SubMatVariable(g, 0,
k,
length(), vn->
width()));
00131
k += vn->
width();
00132 }
00133 }
00134
00135
00136
00137 }
00138
00139