diff options
author | nicholas-leonard <nick@nikopia.org> | 2014-07-09 02:33:19 +0400 |
---|---|---|
committer | nicholas-leonard <nick@nikopia.org> | 2014-07-09 02:33:19 +0400 |
commit | 037c8cdbac310be57ffb921492daf286806020bc (patch) | |
tree | 4f6c36bcbae256f3e97114fb65843be47f62cf47 /generic | |
parent | 11e4e58f22700b8fb4ed033243cb74b0bc14dd53 (diff) |
SoftMaxTree.gradInput is a table (like input)
Diffstat (limited to 'generic')
-rw-r--r-- | generic/SoftMaxTree.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/generic/SoftMaxTree.c b/generic/SoftMaxTree.c index dd4d391..9e66a58 100644 --- a/generic/SoftMaxTree.c +++ b/generic/SoftMaxTree.c @@ -127,7 +127,7 @@ static int nn_(SoftMaxTree_updateGradInput)(lua_State *L) THTensor *weight = luaT_getfieldcheckudata(L, 1, "weight", torch_Tensor); THTensor *output = luaT_getfieldcheckudata(L, 1, "output", torch_Tensor); - THTensor *gradInput = luaT_getfieldcheckudata(L, 1, "gradInput", torch_Tensor); + THTensor *gradInput = luaT_getfieldcheckudata(L, 1, "_gradInput", torch_Tensor); THIntTensor *node; THTensor *nodeWeight, *nodeOutput; |