Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/clementfarabet/lua---nnx.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'generic/SoftMaxTree.c')
-rw-r--r--generic/SoftMaxTree.c7
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");