Browse Source

return EagerTensor directly.

tags/v0.20
Oceania2018 5 years ago
parent
commit
632788086d
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      src/TensorFlowNET.Core/Eager/EagerTensor.cs

+ 3
- 2
src/TensorFlowNET.Core/Eager/EagerTensor.cs View File

@@ -14,8 +14,9 @@ namespace Tensorflow.Eager

public EagerTensor(IntPtr handle) : base(handle)
{
tfe_tensor_handle = handle;
_handle = c_api.TFE_TensorHandleResolve(handle, status);
EagerTensorHandle = handle;
tfe_tensor_handle = c_api.EagerTensor_Handle(handle);
_handle = c_api.TFE_TensorHandleResolve(tfe_tensor_handle, status);
}

public EagerTensor(string value, string device_name) : base(value)


Loading…
Cancel
Save