Merge pull request #5707 from qingqing01/cos_sim_op

qingqing01 · web-flow · commit 9cf603653303 · 2017-11-16T18:57:59.000+08:00
Fix cos_sim_op in debug mode.
diff --git a/paddle/operators/cos_sim_op.h b/paddle/operators/cos_sim_op.h
@@ -132,7 +132,7 @@ class CosSimGradKernel : public framework::OpKernel<T> {
       // compute dy
       if (out_grad_y) {
         out_grad_y->mutable_data<T>(context.GetPlace());
-        auto dy = EigenMatrix<T>::Reshape(*out_grad_y, 1);
+        auto dy = EigenVector<T>::Flatten(*out_grad_y);
         auto grad = x / norm_prod_bcast - z_bcast * y_bcast / y_snorm_bcast;
         dy.device(place) = (dz_bcast * grad).sum(Eigen::array<int, 1>({{0}}));
       }

Original file line number	Diff line number	Diff line change
`@@ -132,7 +132,7 @@ class CosSimGradKernel : public framework::OpKernel<T> {`
`132`	`132`	`// compute dy`
`133`	`133`	`if (out_grad_y) {`
`134`	`134`	`out_grad_y->mutable_data<T>(context.GetPlace());`
`135`		`- auto dy = EigenMatrix<T>::Reshape(*out_grad_y, 1);`
	`135`	`+ auto dy = EigenVector<T>::Flatten(*out_grad_y);`
`136`	`136`	`auto grad = x / norm_prod_bcast - z_bcast * y_bcast / y_snorm_bcast;`
`137`	`137`	`dy.device(place) = (dz_bcast * grad).sum(Eigen::array<int, 1>({{0}}));`
`138`	`138`	`}`