diff options
Diffstat (limited to 'lib/THCUNN/generic/SparseLinear.cu')
-rw-r--r-- | lib/THCUNN/generic/SparseLinear.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/SparseLinear.cu b/lib/THCUNN/generic/SparseLinear.cu index 70c9f5b..07eda62 100644 --- a/lib/THCUNN/generic/SparseLinear.cu +++ b/lib/THCUNN/generic/SparseLinear.cu @@ -206,7 +206,7 @@ void THNN_(SparseLinear_accGradParameters)( &one, THCTensor_(data)(state, gradWeight), inDim ); - THCTensor_(sum)(state, buf, gradOutput, 0); + THCTensor_(sum)(state, buf, gradOutput, 0, 1); THCTensor_(resize1d)(state, buf, outDim); THCTensor_(cadd)(state, gradBias, gradBias, scale, buf); |