From abdf4d815b0cf15fd423acee9123ee8dedd14d81 Mon Sep 17 00:00:00 2001
From: Oceania2018 <haiping008@gmail.com>
Date: Fri, 2 Aug 2019 06:41:54 -0500
Subject: [PATCH] add DisposableObject as base class.

---
 src/TensorFlowNET.Core/Sessions/BaseSession.cs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/TensorFlowNET.Core/Sessions/BaseSession.cs b/src/TensorFlowNET.Core/Sessions/BaseSession.cs
index 05c593c8..21b0a0a1 100644
--- a/src/TensorFlowNET.Core/Sessions/BaseSession.cs
+++ b/src/TensorFlowNET.Core/Sessions/BaseSession.cs
@@ -48,6 +48,10 @@ namespace Tensorflow
 
             _session = c_api.TF_NewSession(_graph, opts ?? newOpts, status);
 
+            // dispose newOpts
+            if (opts == null)
+                c_api.TF_DeleteSessionOptions(newOpts);
+
             status.Check(true);
         }
 
@@ -373,5 +377,19 @@ namespace Tensorflow
                 status.Check(true);
             }
         }
+
+        public void close()
+        {
+            Dispose();
+        }
+
+        protected override void DisposeUnManagedState()
+        {
+            using (var status = new Status())
+            {
+                c_api.TF_DeleteSession(_handle, status);
+                status.Check(true);
+            }
+        }
     }
 }