Merge pull request #1628 from bstatcomp/generalize_operators

Generalize operator-like functions

Author: t4c1

stan/math/fwd/fun.hpp

@@ -23,7 +23,6 @@
 #include <stan/math/fwd/fun/determinant.hpp>
 #include <stan/math/fwd/fun/digamma.hpp>
 #include <stan/math/fwd/fun/divide.hpp>
-#include <stan/math/fwd/fun/dot_product.hpp>
 #include <stan/math/fwd/fun/dot_self.hpp>
 #include <stan/math/fwd/fun/Eigen_NumTraits.hpp>
 #include <stan/math/fwd/fun/erf.hpp>

stan/math/fwd/fun/columns_dot_product.hpp

@@ -16,8 +16,8 @@ inline Eigen::Matrix<fvar<T>, 1, C1> columns_dot_product(
   check_matching_dims("columns_dot_product", "v1", v1, "v2", v2);
   Eigen::Matrix<fvar<T>, 1, C1> ret(1, v1.cols());
   for (size_type j = 0; j < v1.cols(); ++j) {
-    Eigen::Matrix<fvar<T>, R1, C1> ccol1 = v1.col(j);
-    Eigen::Matrix<fvar<T>, R2, C2> ccol2 = v2.col(j);
+    Eigen::Matrix<fvar<T>, R1, 1> ccol1 = v1.col(j);
+    Eigen::Matrix<fvar<T>, R2, 1> ccol2 = v2.col(j);
     ret(0, j) = dot_product(ccol1, ccol2);
   }
   return ret;
@@ -30,8 +30,8 @@ inline Eigen::Matrix<fvar<T>, 1, C1> columns_dot_product(
   check_matching_dims("columns_dot_product", "v1", v1, "v2", v2);
   Eigen::Matrix<fvar<T>, 1, C1> ret(1, v1.cols());
   for (size_type j = 0; j < v1.cols(); ++j) {
-    Eigen::Matrix<fvar<T>, R1, C1> ccol1 = v1.col(j);
-    Eigen::Matrix<double, R2, C2> ccol = v2.col(j);
+    Eigen::Matrix<fvar<T>, R1, 1> ccol1 = v1.col(j);
+    Eigen::Matrix<double, R2, 1> ccol = v2.col(j);
     ret(0, j) = dot_product(ccol1, ccol);
   }
   return ret;
@@ -44,8 +44,8 @@ inline Eigen::Matrix<fvar<T>, 1, C1> columns_dot_product(
   check_matching_dims("columns_dot_product", "v1", v1, "v2", v2);
   Eigen::Matrix<fvar<T>, 1, C1> ret(1, v1.cols());
   for (size_type j = 0; j < v1.cols(); ++j) {
-    Eigen::Matrix<double, R1, C1> ccol = v1.col(j);
-    Eigen::Matrix<fvar<T>, R2, C2> ccol2 = v2.col(j);
+    Eigen::Matrix<double, R1, 1> ccol = v1.col(j);
+    Eigen::Matrix<fvar<T>, R2, 1> ccol2 = v2.col(j);
     ret(0, j) = dot_product(ccol, ccol2);
   }
   return ret;

stan/math/fwd/fun/columns_dot_self.hpp

@@ -3,6 +3,7 @@
 
 #include <stan/math/prim/fun/Eigen.hpp>
 #include <stan/math/fwd/core.hpp>
+#include <stan/math/fwd/fun/typedefs.hpp>
 #include <stan/math/fwd/fun/dot_self.hpp>
 
 namespace stan {

stan/math/fwd/fun/dot_product.hpp

@@ -4,7 +4,7 @@
 #include <stan/math/prim/err.hpp>
 #include <stan/math/prim/fun/Eigen.hpp>
 #include <stan/math/fwd/core.hpp>
-#include <stan/math/fwd/fun/dot_product.hpp>
+#include <stan/math/prim/fun/dot_product.hpp>
 
 namespace stan {
 namespace math {