tf.zeros for dtype uint8

Oceania2018 · Oceania2018 · commit 99fc01611e21 · 2021-07-20T12:44:35.000-05:00
diff --git a/src/TensorFlowNET.Core/Data/MnistModelLoader.cs b/src/TensorFlowNET.Core/Data/MnistModelLoader.cs
@@ -123,7 +123,7 @@ private NDArray ExtractImages(string file, int? limit = null)
 
                 bytestream.Read(buf, 0, buf.Length);
 
-                var data = np.frombuffer(buf, (num_images, rows * cols), np.@byte);
+                var data = np.frombuffer(buf, (num_images, rows * cols), np.uint8);
                 return data;
             }
         }
@@ -146,7 +146,7 @@ private NDArray ExtractLabels(string file, bool one_hot = false, int num_classes
 
                 bytestream.Read(buf, 0, buf.Length);
 
-                var labels = np.frombuffer(buf, new Shape(num_items), np.@byte);
+               var labels = np.frombuffer(buf, new Shape(num_items), np.uint8);
 
                 if (one_hot)
                     return DenseToOneHot(labels, num_classes);
diff --git a/src/TensorFlowNET.Core/Operations/array_ops.cs b/src/TensorFlowNET.Core/Operations/array_ops.cs
@@ -91,6 +91,9 @@ public static Tensor zeros(Shape shape, TF_DataType dtype = TF_DataType.TF_FLOAT
                             zeros = constant(0f);
                             break;
                         case TF_DataType.TF_INT8:
+                            zeros = constant((sbyte)0);
+                            break;
+                        case TF_DataType.TF_UINT8:
                             zeros = constant((byte)0);
                             break;
                         default:

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ private NDArray ExtractImages(string file, int? limit = null)`
`123`	`123`
`124`	`124`	`bytestream.Read(buf, 0, buf.Length);`
`125`	`125`
`126`		`- var data = np.frombuffer(buf, (num_images, rows * cols), np.@byte);`
	`126`	`+ var data = np.frombuffer(buf, (num_images, rows * cols), np.uint8);`
`127`	`127`	`return data;`
`128`	`128`	`}`
`129`	`129`	`}`
`@@ -146,7 +146,7 @@ private NDArray ExtractLabels(string file, bool one_hot = false, int num_classes`
`146`	`146`
`147`	`147`	`bytestream.Read(buf, 0, buf.Length);`
`148`	`148`
`149`		`- var labels = np.frombuffer(buf, new Shape(num_items), np.@byte);`
	`149`	`+ var labels = np.frombuffer(buf, new Shape(num_items), np.uint8);`
`150`	`150`
`151`	`151`	`if (one_hot)`
`152`	`152`	`return DenseToOneHot(labels, num_classes);`