add begin/end to ExecuTorch pytree::arr

extension/pytree/aten_util/ivalue_util.cpp

@@ -131,8 +131,8 @@ std::pair<std::vector<at::Tensor>, std::unique_ptr<TreeSpec<Empty>>> flatten(
   auto p = flatten(c);
 
   std::vector<at::Tensor> tensors;
-  for (int i = 0; i < p.first.size(); ++i) {
-    tensors.emplace_back(p.first[i]->toTensor());
+  for (const auto& item : p.first) {
+    tensors.emplace_back(item->toTensor());
   }
 
   return {tensors, std::move(p.second)};

extension/pytree/pytree.h

@@ -431,6 +431,22 @@ struct arr {
     return data_.get();
   }
 
+  T* begin() {
+    return data_.get();
+  }
+
+  T* end() {
+    return begin() + size();
+  }
+
+  const T* begin() const {
+    return data_.get();
+  }
+
+  const T* end() const {
+    return begin() + size();
+  }
+
   inline size_t size() const {
     return n_;
   }