fixed Tensor memory leak.

Oceania2018 · Oceania2018 · commit 2f29995d146f · 2019-03-27T15:13:45.000-05:00
diff --git a/src/TensorFlowNET.Core/Sessions/BaseSession.cs b/src/TensorFlowNET.Core/Sessions/BaseSession.cs
@@ -186,9 +186,10 @@ private unsafe NDArray[] _call_tf_sessionrun(KeyValuePair<TF_Output, Tensor>[] f
             var result = new NDArray[fetch_list.Length];
 
             for (int i = 0; i < fetch_list.Length; i++)
-            {
                 result[i] = fetchValue(output_values[i]);
-            }
+
+            for (int i = 0; i < feed_dict.Length; i++)
+                feed_dict[i].Value.Dispose();
 
             return result;
         }
diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
@@ -111,7 +111,7 @@ private IntPtr Allocate(NDArray nd)
             // Free the original buffer and set flag
             Deallocator deallocator = (IntPtr values, IntPtr len, ref bool closure) =>
             {
-                Marshal.FreeHGlobal(dotHandle);
+                Marshal.FreeHGlobal(values);
                 closure = true;
             };
 

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,7 @@ private IntPtr Allocate(NDArray nd)`
`111`	`111`	`// Free the original buffer and set flag`
`112`	`112`	`Deallocator deallocator = (IntPtr values, IntPtr len, ref bool closure) =>`
`113`	`113`	`{`
`114`		`- Marshal.FreeHGlobal(dotHandle);`
	`114`	`+ Marshal.FreeHGlobal(values);`
`115`	`115`	`closure = true;`
`116`	`116`	`};`
`117`	`117`