refine seq2batch

paddle/fluid/operators/math/sequence2batch.cc

@@ -38,13 +38,14 @@ class CopyMatrixRowsFunctor<platform::CPUDeviceContext, T> {
     auto width = dst_dims[1];
     auto* src_data = src.data<T>();
     auto* dst_data = dst->data<T>();
-    for (int i = 0; i < height; ++i) {
+    const int sz = width * sizeof(T);
-      if (is_src_index) {
+    if (is_src_index) {
-        memcpy(dst_data + i * width, src_data + index[i] * width,
+      for (int i = 0; i < height; ++i) {
-               width * sizeof(T));
+        memcpy(dst_data + i * width, src_data + index[i] * width, sz);
-      } else {
+      }
-        memcpy(dst_data + index[i] * width, src_data + i * width,
+    } else {
-               width * sizeof(T));
+      for (int i = 0; i < height; ++i) {
+        memcpy(dst_data + index[i] * width, src_data + i * width, sz);
       }
     }
   }