diff options
Diffstat (limited to 'src/core/ITensorPack.cpp')
-rw-r--r-- | src/core/ITensorPack.cpp | 5 |
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(); |