Browse Source

Tensor: added missing data types

tags/v0.9
Meinrad Recheis 6 years ago
parent
commit
5d16dc67cb
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      src/TensorFlowNET.Core/Tensors/Tensor.cs

+ 4
- 0
src/TensorFlowNET.Core/Tensors/Tensor.cs View File

@@ -191,6 +191,8 @@ namespace Tensorflow
return TF_DataType.TF_INT16;
case "Int32":
return TF_DataType.TF_INT32;
case "Int64":
return TF_DataType.TF_INT64;
case "Single":
return TF_DataType.TF_FLOAT;
case "Double":
@@ -199,6 +201,8 @@ namespace Tensorflow
return TF_DataType.TF_UINT8;
case "String":
return TF_DataType.TF_STRING;
case "Boolean":
return TF_DataType.TF_BOOL;
default:
throw new NotImplementedException("ToTFDataType error");
}


Loading…
Cancel
Save