addr zeroes output buffer when beta=0

little1tow · Apr 12, 2017 · 0925c91 · 0925c91
1 parent 662163b
commit 0925c91
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/generic/THTensorMath.c b/generic/THTensorMath.c
@@ -1365,7 +1365,10 @@ void THTensor_(addr)(THTensor *r_, real beta, THTensor *t, real alpha, THTensor
     THTensor_(copy)(r_, t);
   }
 
-  if(beta != 1)
+  if(beta == 0) {
+    THTensor_(zero)(r_);
+  }
+  else if(beta != 1)
     THTensor_(mul)(r_, r_, beta);
 
   if(r_->stride[0] == 1)