diff options
-rw-r--r-- | generic/Square.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/generic/Square.c b/generic/Square.c index 33feaa9..4e20116 100644 --- a/generic/Square.c +++ b/generic/Square.c @@ -40,7 +40,7 @@ static int nn_(Square_updateGradInput)(lua_State *L) !THTensor_(isContiguous)(gradInput)) { TH_TENSOR_APPLY3(real, gradInput, real, gradOutput, real, input, \ - *gradInput_data = (*gradOutput_data) * (*input_data);); + *gradInput_data = 2.0 * (*gradOutput_data) * (*input_data);); } else { |