fix GetShape for Axis.

Oceania2018 · Oceania2018 · commit f9802664d354 · 2021-07-21T23:07:33.000-05:00
diff --git a/src/TensorFlowNET.Core/Binding.Util.cs b/src/TensorFlowNET.Core/Binding.Util.cs
@@ -517,7 +517,7 @@ public static Shape GetShape(this object data)
                 return tensor.shape;
 
             else if (data is Axis axis)
-                return axis.IsScalar ? Shape.Scalar : new Shape(axis.axis);
+                return axis.IsScalar ? Shape.Scalar : new Shape(axis.axis.Length);
 
             else if (!data.GetType().IsArray)
                 return Shape.Scalar;
diff --git a/src/TensorFlowNET.Keras/Utils/np_utils.cs b/src/TensorFlowNET.Keras/Utils/np_utils.cs
@@ -20,9 +20,9 @@ public static NDArray to_categorical(NDArray y, int num_classes = -1, TF_DataTyp
             // var input_shape = y.shape[..^1];
             var categorical = np.zeros(((int)y.size, num_classes), dtype: dtype);
             // categorical[np.arange(y.size), y] = 1;
-            for (ulong i = 0; i < y.size; i++)
+            for (var i = 0; i < (int)y.size; i++)
             {
-                categorical[i][y1[i]] = 1;
+                categorical[i][y1[i]] = 1.0f;
             }
 
             return categorical;

Original file line number	Diff line number	Diff line change
`@@ -20,9 +20,9 @@ public static NDArray to_categorical(NDArray y, int num_classes = -1, TF_DataTyp`
`20`	`20`	`// var input_shape = y.shape[..^1];`
`21`	`21`	`var categorical = np.zeros(((int)y.size, num_classes), dtype: dtype);`
`22`	`22`	`// categorical[np.arange(y.size), y] = 1;`
`23`		`- for (ulong i = 0; i < y.size; i++)`
	`23`	`+ for (var i = 0; i < (int)y.size; i++)`
`24`	`24`	`{`
`25`		`- categorical[i][y1[i]] = 1;`
	`25`	`+ categorical[i][y1[i]] = 1.0f;`
`26`	`26`	`}`
`27`	`27`
`28`	`28`	`return categorical;`