Add getLayers method to ModelT (#158)

jatinchowdhury18 · Dec 5, 2024 · 21ebabd · 21ebabd
1 parent cd80257
commit 21ebabd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/RTNeural/ModelT.h b/RTNeural/ModelT.h
@@ -445,6 +445,12 @@ class ModelT
         return parseJson(parent, debug, custom_layers);
     }
 
+    /** Returns a reference to a tuple containing the model layers */
+    auto& getLayers() noexcept
+    {
+        return layers;
+    }
+
 private:
 #if RTNEURAL_USE_XSIMD
     using v_type = xsimd::simd_type<T>;