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 "AffineTransformVariable.h"
00044
00045
namespace PLearn {
00046
using namespace std;
00047
00048
00049
PLEARN_IMPLEMENT_OBJECT(AffineTransformVariable,
00050
"Affine transformation of a vector variable.",
00051
"NO HELP");
00052
00053 AffineTransformVariable::AffineTransformVariable(
Variable* vec,
Variable* transformation)
00054 :
inherited(vec, transformation,
00055 (vec->size() == 1) ? transformation->width() : (vec->isRowVec() ? 1 : transformation->width()),
00056 (vec->size() == 1) ? 1 : (vec->isRowVec() ? transformation->width() : 1))
00057 {
00058
build_();
00059 }
00060
00061
void
00062 AffineTransformVariable::build()
00063 {
00064 inherited::build();
00065
build_();
00066 }
00067
00068
void
00069 AffineTransformVariable::build_()
00070 {
00071
00072
if (input1 && !input1->isVec())
00073
PLERROR(
"In AffineTransformVariable: expecting a vector Var (row or column) as first argument");
00074 }
00075
00076 void AffineTransformVariable::recomputeSize(
int& l,
int& w)
const
00077
{
00078
if (input1 && input2) {
00079 l = input1->isRowVec() ? 1 : input2->
width();
00080 w = input1->isColumnVec() ? 1 : input2->
width();
00081 }
else
00082 l = w = 0;
00083 }
00084
00085
00086 void AffineTransformVariable::fprop()
00087 {
00088 value << input2->matValue.firstRow();
00089
Mat lintransform = input2->matValue.subMatRows(1,input2->
length()-1);
00090
transposeProductAcc(value, lintransform, input1->value);
00091 }
00092
00093
00094 void AffineTransformVariable::bprop()
00095 {
00096
Mat& afftr = input2->matValue;
00097
int l = afftr.
length();
00098
00099
Mat lintr = afftr.
subMatRows(1,l-1);
00100
00101
Mat& afftr_g = input2->matGradient;
00102
Vec bias_g = afftr_g.
firstRow();
00103
Mat lintr_g = afftr_g.
subMatRows(1,l-1);
00104
00105 bias_g += gradient;
00106
if(!input1->dont_bprop_here)
00107
productAcc(input1->gradient, lintr, gradient);
00108
externalProductAcc(lintr_g, input1->value, gradient);
00109 }
00110
00111
00112 void AffineTransformVariable::symbolicBprop()
00113 {
00114
PLERROR(
"AffineTransformVariable::symbolicBprop() not implemented");
00115 }
00116
00117 }
00118
00119