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 "ConcatOfVariable.h"
00044
00045
namespace PLearn {
00046
using namespace std;
00047
00048
00049
00051
00052
00053
PLEARN_IMPLEMENT_OBJECT(ConcatOfVariable,
"Concatenates the results of each operation in the loop into the resulting variable",
"NO HELP");
00054
00055 ConcatOfVariable::ConcatOfVariable(
VMat the_distr,
Func the_f)
00056 :
inherited(
nonInputParentsOfPath(the_f->inputs, the_f->outputs),
00057 the_f->outputs[0]->length() * the_distr->length(),
00058 the_f->outputs[0]->width()),
00059 distr(the_distr), f(the_f)
00060 {
00061
build_();
00062 }
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
void
00076 ConcatOfVariable::declareOptions(
OptionList &ol)
00077 {
00078
declareOption(ol,
"distr", &ConcatOfVariable::distr, OptionBase::buildoption,
"");
00079
declareOption(ol,
"f", &ConcatOfVariable::f, OptionBase::buildoption,
"");
00080 inherited::declareOptions(ol);
00081 }
00082
00083
void
00084 ConcatOfVariable::build()
00085 {
00086 inherited::build();
00087
build_();
00088 }
00089
00090
void
00091 ConcatOfVariable::build_()
00092 {
00093
if (
distr &&
f) {
00094
if(
f->outputs.size()!=1)
00095
PLERROR(
"In ConcatOfVariable constructor: function must have a single output variable");
00096
00097
input_value.
resize(
distr->
width());
00098
input_gradient.
resize(
distr->
width());
00099 }
00100 }
00101
00102
00103 void ConcatOfVariable::recomputeSize(
int& l,
int& w)
const
00104
{
00105
if (
f &&
distr) {
00106 l =
f->outputs[0]->length() *
distr->
length();
00107 w =
f->outputs[0]->width();
00108 }
else
00109 l = w = 0;
00110 }
00111
00112
00113 void ConcatOfVariable::makeDeepCopyFromShallowCopy(map<const void*, void*>& copies)
00114 {
00115 NaryVariable::makeDeepCopyFromShallowCopy(copies);
00116
deepCopyField(
distr, copies);
00117
deepCopyField(
f, copies);
00118
deepCopyField(
input_value, copies);
00119
deepCopyField(
input_gradient, copies);
00120 }
00121
00122
00123
00124 void ConcatOfVariable::fprop()
00125 {
00126
f->recomputeParents();
00127
00128
int pos = 0;
00129
int singleoutputsize =
f->outputs[0]->nelems();
00130
for(
int i=0; i<
distr->
length(); i++, pos+=singleoutputsize)
00131 {
00132
distr->getRow(i,
input_value);
00133
f->fprop(
input_value, value.
subVec(pos,singleoutputsize));
00134 }
00135 }
00136
00137
00138 void ConcatOfVariable::bprop()
00139 {
00140
fbprop();
00141 }
00142
00143
00144 void ConcatOfVariable::fbprop()
00145 {
00146
f->recomputeParents();
00147
00148
int pos = 0;
00149
int singleoutputsize =
f->outputs[0]->nelems();
00150
for(
int i=0; i<
distr->
length(); i++, pos+=singleoutputsize)
00151 {
00152
distr->getRow(i,
input_value);
00153
f->fbprop(
input_value, value.
subVec(pos,singleoutputsize),
00154
input_gradient, gradient.
subVec(pos,singleoutputsize));
00155
00156
00157
00158 }
00159 }
00160
00161
00162
00163 }
00164
00165