diff options
Diffstat (limited to 'generic/SoftMaxTree.c')
-rw-r--r-- | generic/SoftMaxTree.c | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/generic/SoftMaxTree.c b/generic/SoftMaxTree.c index 9e66a58..074c74e 100644 --- a/generic/SoftMaxTree.c +++ b/generic/SoftMaxTree.c @@ -126,13 +126,12 @@ static int nn_(SoftMaxTree_updateGradInput)(lua_State *L) THTensor *logsoftOutput = luaT_getfieldcheckudata(L, 1, "_multiBuffer", torch_Tensor); 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); THIntTensor *node; THTensor *nodeWeight, *nodeOutput; - THTensor *nodeGradInput, *nodeGradOutput, *weightTranspose; - real *gradInput_data, *output_data; + THTensor *nodeGradInput, *weightTranspose; + real *output_data; long i, d; @@ -231,7 +230,7 @@ static int nn_(SoftMaxTree_accGradParameters)(lua_State *L) THIntTensor *node; THTensor *nodeGradWeight, *nodeGradBias, *nodeInput, *nodeGradOutput; - long i, d; + long i; luaL_argcheck(L, input->nDimension == 2, 2, "2D(batch mode) tensor expected"); luaL_argcheck(L, input->size[1] == inputSize, 2, "invalid input size"); |