Browse Source

Tensor.Creation.CreateTensorFromNDArray: Fixed AllocationReferenceHolder when sliced.

tags/v0.12
Eli Belash 6 years ago
parent
commit
2b9bd5fb8c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs

+ 1
- 1
src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs View File

@@ -523,7 +523,7 @@ namespace Tensorflow
if (TF_TensorData(handle).ToPointer() == arraySlice.Address)
{
AllocationType = AllocationType.FromPointer;
AllocationReferenceHolder = nd.Unsafe.Storage;
AllocationReferenceHolder = arraySlice;
} else
AllocationType = AllocationType.Tensorflow;



Loading…
Cancel
Save