Fix sparse_categorical_crossentropy.

Oceania2018 · Oceania2018 · commit 9aa8f758f402 · 2023-03-02T21:57:48.000-06:00
diff --git a/src/TensorFlowNET.Keras/BackendImpl.cs b/src/TensorFlowNET.Keras/BackendImpl.cs
@@ -307,9 +307,8 @@ public Tensor sparse_categorical_crossentropy(Tensor target, Tensor output, bool
             var update_shape = target_rank > -1 && output_rank > -1 && target_rank != output_rank - 1;
             if (update_shape)
             {
-                /*var target = flatten(target);
-                output = tf.reshape(output, [-1, output_shape[-1]]);*/
-                throw new NotImplementedException("");
+                target = tf.reshape(target, -1);
+                output = tf.reshape(output, (-1, output.shape[-1]));
             }
 
             if (ignore_class.HasValue)
diff --git a/test/TensorFlowNET.Keras.UnitTest/MultiThreadsTest.cs b/test/TensorFlowNET.Keras.UnitTest/MultiThreadsTest.cs
@@ -8,7 +8,7 @@
 
 namespace TensorFlowNET.Keras.UnitTest
 {
-    [TestClass, Ignore]
+    [TestClass]
     public class MultiThreads
     {
         [TestMethod]

Original file line number	Diff line number	Diff line change
`@@ -307,9 +307,8 @@ public Tensor sparse_categorical_crossentropy(Tensor target, Tensor output, bool`
`307`	`307`	`var update_shape = target_rank > -1 && output_rank > -1 && target_rank != output_rank - 1;`
`308`	`308`	`if (update_shape)`
`309`	`309`	`{`
`310`		`- /*var target = flatten(target);`
`311`		`- output = tf.reshape(output, [-1, output_shape[-1]]);*/`
`312`		`- throw new NotImplementedException("");`
	`310`	`+ target = tf.reshape(target, -1);`
	`311`	`+ output = tf.reshape(output, (-1, output.shape[-1]));`
`313`	`312`	`}`
`314`	`313`
`315`	`314`	`if (ignore_class.HasValue)`
Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`
`9`	`9`	`namespace TensorFlowNET.Keras.UnitTest`
`10`	`10`	`{`
`11`		`- [TestClass, Ignore]`
	`11`	`+ [TestClass]`
`12`	`12`	`public class MultiThreads`
`13`	`13`	`{`
`14`	`14`	`[TestMethod]`