There was an error while loading. Please reload this page.
2 parents 194d740 + 906c550 commit c34cff7Copy full SHA for c34cff7
torch/lib/THNN/generic/SparseLinear.c
@@ -234,7 +234,7 @@ void THNN_(SparseLinear_accGradParameters)(
234
235
// gradBias += gradOutput
236
THTensor* buf = THTensor_(new)();
237
- THTensor_(sum)(buf, gradOutput, 0);
+ THTensor_(sum)(buf, gradOutput, 0, 1);
238
THTensor_(cadd)(gradBias, gradBias, scale, buf);
239
THTensor_(free)(buf);
240
THLongTensor_free(csc);
torch/lib/THNN/generic/TemporalSubSampling.c
@@ -70,7 +70,7 @@ void THNN_(TemporalSubSampling_updateOutput)(
70
{
71
THTensor_(narrow)(inputWindow, input, 0, k*dW, kW);
72
THTensor_(select)(outputFrame, output, 0, k);
73
- THTensor_(sum)(outputFrame, inputWindow, 0);
+ THTensor_(sum)(outputFrame, inputWindow, 0, 1);
74
THTensor_(cmul)(outputFrame, outputFrame, weight);
75
THTensor_(cadd)(outputFrame, outputFrame, 1, bias);
76
}
@@ -143,7 +143,7 @@ void THNN_(TemporalSubSampling_accGradParameters)(
143
144
145
THTensor_(select)(gradOutputFrame, gradOutput, 0, k);
146
- THTensor_(sum)(buffer, inputWindow, 0);
+ THTensor_(sum)(buffer, inputWindow, 0, 1);
147
THTensor_(addcmul)(gradWeight, gradWeight, scale, buffer, gradOutputFrame);
148
THTensor_(cadd)(gradBias, gradBias, scale, gradOutputFrame);
149
0 commit comments