diff options
Diffstat (limited to 'generic/SpatialLinear.c')
-rw-r--r-- | generic/SpatialLinear.c | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/generic/SpatialLinear.c b/generic/SpatialLinear.c index 8a8b756..113eb85 100644 --- a/generic/SpatialLinear.c +++ b/generic/SpatialLinear.c @@ -105,6 +105,7 @@ static int nn_(SpatialLinear_updateGradInput)(lua_State *L) THTensor_(free)(gradOutput_y); THTensor_(free)(input_xy); THTensor_(free)(input_y); + THTensor_(free)(weight_t); return 1; } |