1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/ITensorPack.cpp b/src/core/ITensorPack.cpp
index 9eaeece271..90f9a45039 100644
--- a/src/core/ITensorPack.cpp
+++ b/src/core/ITensorPack.cpp
@@ -67,6 +67,11 @@ ITensor *ITensorPack::get_tensor(int id)
     return it != _pack.end() ? it->second.tensor : nullptr;
 }
 
+void ITensorPack::remove_tensor(int id)
+{
+    _pack.erase(id);
+}
+
 size_t ITensorPack::size() const
 {
     return _pack.size();