Browse Source

fix bool warning

tags/v0.9
Oceania2018 6 years ago
parent
commit
6845233c09
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs

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

@@ -56,7 +56,8 @@ namespace Tensorflow
switch (nd.dtype.Name)
{
case "Boolean":
Marshal.Copy(nd1.Data<byte>(), 0, dotHandle, nd.size);
var boolVals = Array.ConvertAll(nd1.Data<bool>(), x => Convert.ToByte(x));
Marshal.Copy(boolVals, 0, dotHandle, nd.size);
break;
case "Int16":
Marshal.Copy(nd1.Data<short>(), 0, dotHandle, nd.size);


Loading…
Cancel
Save