Skip to content

Commit c34cff7

Browse files
committed
Merge commit '906c550e1079e9762194db59440a202ffca90dca'
2 parents 194d740 + 906c550 commit c34cff7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

torch/lib/THNN/generic/SparseLinear.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ void THNN_(SparseLinear_accGradParameters)(
234234

235235
// gradBias += gradOutput
236236
THTensor* buf = THTensor_(new)();
237-
THTensor_(sum)(buf, gradOutput, 0);
237+
THTensor_(sum)(buf, gradOutput, 0, 1);
238238
THTensor_(cadd)(gradBias, gradBias, scale, buf);
239239
THTensor_(free)(buf);
240240
THLongTensor_free(csc);

torch/lib/THNN/generic/TemporalSubSampling.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ void THNN_(TemporalSubSampling_updateOutput)(
7070
{
7171
THTensor_(narrow)(inputWindow, input, 0, k*dW, kW);
7272
THTensor_(select)(outputFrame, output, 0, k);
73-
THTensor_(sum)(outputFrame, inputWindow, 0);
73+
THTensor_(sum)(outputFrame, inputWindow, 0, 1);
7474
THTensor_(cmul)(outputFrame, outputFrame, weight);
7575
THTensor_(cadd)(outputFrame, outputFrame, 1, bias);
7676
}
@@ -143,7 +143,7 @@ void THNN_(TemporalSubSampling_accGradParameters)(
143143
{
144144
THTensor_(narrow)(inputWindow, input, 0, k*dW, kW);
145145
THTensor_(select)(gradOutputFrame, gradOutput, 0, k);
146-
THTensor_(sum)(buffer, inputWindow, 0);
146+
THTensor_(sum)(buffer, inputWindow, 0, 1);
147147
THTensor_(addcmul)(gradWeight, gradWeight, scale, buffer, gradOutputFrame);
148148
THTensor_(cadd)(gradBias, gradBias, scale, gradOutputFrame);
149149
}

0 commit comments

Comments
 (0)