Browse Source

convert_to_eager_tensor

tags/v0.60-tf.numpy
Oceania2018 4 years ago
parent
commit
9849829d41
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Tensors/Tensor.Value.cs
  2. +2
    -0
      src/TensorFlowNET.Core/Tensors/constant_op.cs

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

@@ -52,7 +52,7 @@ namespace Tensorflow
if (dtype == TF_DataType.TF_STRING) if (dtype == TF_DataType.TF_STRING)
{ {
var str= StringData(); var str= StringData();
return new NDArray(str, new Shape(str.Length));
return new NDArray(str, shape);
} }
return new NDArray(this); return new NDArray(this);


+ 2
- 0
src/TensorFlowNET.Core/Tensors/constant_op.cs View File

@@ -136,6 +136,8 @@ namespace Tensorflow
return new EagerTensor(new[] { val }, Shape.Scalar); return new EagerTensor(new[] { val }, Shape.Scalar);
case double val: case double val:
return new EagerTensor(new[] { val }, Shape.Scalar); return new EagerTensor(new[] { val }, Shape.Scalar);
case IEnumerable<Tensor> val:
return ops.convert_to_tensor(val);
case Array val: case Array val:
return new EagerTensor(val, val.GetShape()); return new EagerTensor(val, val.GetShape());
default: default:


Loading…
Cancel
Save