fix a bug

7 years ago · 12d17941a1
parent 41ced8e2af
commit 12d17941a1
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/reverse_op.h
+++ b/paddle/fluid/operators/reverse_op.h
@ -24,8 +24,8 @@ struct ReverseFunctor {
  void operator()(const DeviceContext& context, const framework::LoDTensor& in,
                  framework::LoDTensor* out, const std::vector<int>& axis) {
    Eigen::array<bool, Rank> reverse_axis;
-    for (auto& i : reverse_axis) {
-      i = false;
+    for (int i = 0; i < Rank; ++i) {
+      reverse_axis[i] = false;
    }
    for (int a : axis) {
      reverse_axis[a] = true;