Tensor.Creation: Added IntPtr.Zero check for FreeHGlobalMemory

SciSharp · Aug 30, 2019 · 1d6f3de · 1d6f3de
1 parent 288085b
commit 1d6f3de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
@@ -664,7 +664,7 @@ protected unsafe IntPtr CreateTensorWithoutCopying(TF_DataType dt, long[] shape,
         [MonoPInvokeCallback(typeof(Deallocator))]
         internal static void FreeHGlobalMemory(IntPtr dataPtr, IntPtr len, ref DeallocatorArgs args)
         {
-            if (args.deallocator_called)
+            if (args.deallocator_called || dataPtr == IntPtr.Zero)
                 return;
 
             // NumSharp will dispose