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

softmax grad

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-18 19:45:02 +01:00
parent 022a20efb7
commit 40a9791f69
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/test_nodes.cu
View File

@ -34,6 +34,9 @@ int main(int argc, char** argv)
inputExpr = inputTensor;
Expr softMaxExpr = softmax(inputExpr);
Tensor softMaxGrad = softMaxExpr.grad();
std::cerr << "softMaxGrad=" << softMaxGrad.Debug() << std::endl;
g.forward(batch_size);
g.backward();