fix tensor rm method bug

sipeed · Aug 7, 2024 · ae9435b · ae9435b
1 parent 11af2ff
commit ae9435b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/components/basic/include/maix_tensor.hpp b/components/basic/include/maix_tensor.hpp
@@ -491,6 +491,10 @@ namespace maix
             */
             void rm_tensor(const std::string &key)
             {
+                if(_auto_delete[key])
+                {
+                    delete tensors[key];
+                }
                 tensors.erase(key);
             }