Tensor.Creation.CreateTensorFromNDArray: Fixed AllocationReferenceHolder when sliced.

Nucs · Nucs · commit 2b9bd5fb8c32 · 2019-08-31T16:47:43.000+03:00
diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
@@ -523,7 +523,7 @@ private unsafe IntPtr CreateTensorFromNDArray(NDArray nd, TF_DataType? given_dty
             if (TF_TensorData(handle).ToPointer() == arraySlice.Address)
             {
                 AllocationType = AllocationType.FromPointer;
-                AllocationReferenceHolder = nd.Unsafe.Storage;
+                AllocationReferenceHolder = arraySlice;
             } else
                 AllocationType = AllocationType.Tensorflow;
 

Original file line number	Diff line number	Diff line change
`@@ -523,7 +523,7 @@ private unsafe IntPtr CreateTensorFromNDArray(NDArray nd, TF_DataType? given_dty`
`523`	`523`	`if (TF_TensorData(handle).ToPointer() == arraySlice.Address)`
`524`	`524`	`{`
`525`	`525`	`AllocationType = AllocationType.FromPointer;`
`526`		`- AllocationReferenceHolder = nd.Unsafe.Storage;`
	`526`	`+ AllocationReferenceHolder = arraySlice;`
`527`	`527`	`} else`
`528`	`528`	`AllocationType = AllocationType.Tensorflow;`
`529`	`529`