Optimize imdb.load_data

SciSharp · Sep 18, 2023 · 725ec1e · 725ec1e
1 parent 74afa14
commit 725ec1e
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/TensorFlowNET.Keras/Datasets/Imdb.cs b/src/TensorFlowNET.Keras/Datasets/Imdb.cs
@@ -180,10 +180,11 @@ public DatasetPass load_data(
             // 0 (padding), 1 (start), 2 (OOV)
             if (oov_char != null)
             {
-                int[,] new_xs_array = new int[xs_array.GetLength(0), xs_array.GetLength(1)];
-                for (var i = 0; i < xs_array.GetLength(0); i++)
+                var (d1, d2) = (xs_array.GetLength(0), xs_array.GetLength(1));
+                int[,] new_xs_array = new int[d1, d2];
+                for (var i = 0; i < d1; i++)
                 {
-                    for (var j = 0; j < xs_array.GetLength(1); j++)
+                    for (var j = 0; j < d2; j++)
                     {
                         if (xs_array[i, j] == 0 || skip_top <= xs_array[i, j] && xs_array[i, j] < num_words)
                             new_xs_array[i, j] = xs_array[i, j];
@@ -195,11 +196,12 @@ public DatasetPass load_data(
             }
             else
             {
-                int[,] new_xs_array = new int[xs_array.GetLength(0), xs_array.GetLength(1)];
-                for (var i = 0; i < xs_array.GetLength(0); i++)
+                var (d1, d2) = (xs_array.GetLength(0), xs_array.GetLength(1));
+                int[,] new_xs_array = new int[d1, d2];
+                for (var i = 0; i < d1; i++)
                 {
                     int k = 0;
-                    for (var j = 0; j < xs_array.GetLength(1); j++)
+                    for (var j = 0; j < d2; j++)
                     {
                         if (xs_array[i, j] == 0 || skip_top <= xs_array[i, j] && xs_array[i, j] < num_words)
                             new_xs_array[i, k++] = xs_array[i, j];