marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

re-arrange expressions to test them all

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-21 19:49:07 +01:00
parent 3a8659c924
commit fbe0126c07
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/test_nodes.cu
View File

@ -34,21 +34,21 @@ int main(int argc, char** argv)
vector<Expr> expr;
expr.emplace_back(tanh(inExpr));
expr.emplace_back(inExpr + inExpr2);
expr.emplace_back(inExpr - expr.back());
expr.emplace_back(inExpr * expr.back());
expr.emplace_back(inExpr / expr.back());
expr.emplace_back(reluplus(inExpr, expr.back()));
//expr.emplace_back(dot(inExpr, inExpr3));
expr.emplace_back(tanh(expr.back()));
expr.emplace_back(-expr.back());
expr.emplace_back(logit(expr.back()));
expr.emplace_back(relu(expr.back()));
expr.emplace_back(log(expr.back()));
expr.emplace_back(exp(expr.back()));
expr.emplace_back(softmax(expr.back()));
expr.emplace_back(inExpr + inExpr2);
expr.emplace_back(inExpr - inExpr2);
expr.emplace_back(inExpr * inExpr2);
expr.emplace_back(inExpr / inExpr2);
expr.emplace_back(reluplus(inExpr, inExpr2));
//expr.emplace_back(dot(inExpr, inExpr3));
//expr.emplace_back(softmax(expr.back()));
Expr ceExpr = cross_entropy(expr.back(), labelExpr);
Expr cost = mean(ceExpr, axis=0);