From 99fc01611e218422ff4ceb587e9b05d42e46d43d Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Tue, 20 Jul 2021 12:44:35 -0500 Subject: [PATCH] tf.zeros for dtype uint8 --- src/TensorFlowNET.Core/Data/MnistModelLoader.cs | 4 ++-- src/TensorFlowNET.Core/Operations/array_ops.cs | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Core/Data/MnistModelLoader.cs b/src/TensorFlowNET.Core/Data/MnistModelLoader.cs index 0aaeb32a..50acb4ba 100644 --- a/src/TensorFlowNET.Core/Data/MnistModelLoader.cs +++ b/src/TensorFlowNET.Core/Data/MnistModelLoader.cs @@ -123,7 +123,7 @@ namespace Tensorflow 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 @@ namespace Tensorflow 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 index 287bc37a..26bbd042 100644 --- a/src/TensorFlowNET.Core/Operations/array_ops.cs +++ b/src/TensorFlowNET.Core/Operations/array_ops.cs @@ -91,6 +91,9 @@ namespace Tensorflow 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: