From e92b0bd8c2eec33061581ef45dc8ecac8c6330da Mon Sep 17 00:00:00 2001
From: Shangtong Zhang <zhangshangtong.cpp@icloud.com>
Date: Thu, 22 Jun 2017 20:12:08 -0600
Subject: [PATCH] Update Add

---
 src/mlpack/methods/ann/layer/add_impl.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/mlpack/methods/ann/layer/add_impl.hpp b/src/mlpack/methods/ann/layer/add_impl.hpp
index 3ce562007a..b04bb8247f 100644
--- a/src/mlpack/methods/ann/layer/add_impl.hpp
+++ b/src/mlpack/methods/ann/layer/add_impl.hpp
@@ -31,7 +31,8 @@ template<typename eT>
 void Add<InputDataType, OutputDataType>::Forward(
     const arma::Mat<eT>&& input, arma::Mat<eT>&& output)
 {
-  output = input + weights;
+  output = input;
+  output.each_col() += weights;
 }
 
 template<typename InputDataType, typename OutputDataType>