Improved softmax tests

a4c5983a · Davis King · 0a832e42 · a4c5983a
Commit a4c5983a authored Dec 12, 2015 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

dlib/test/dnn.cpp dlib/test/dnn.cpp +6 -4

No files found.
--- a/dlib/test/dnn.cpp
+++ b/dlib/test/dnn.cpp
@@ -114,10 +114,12 @@ namespace
    void test_softmax()
    {
        print_spinner();
-        resizable_tensor src(5,5), dest(5,5), gradient_input(5,5);
-        src = matrix_cast<float>(gaussian_randm(5,5, 0));
-        dest = matrix_cast<float>(gaussian_randm(5,5, 1));
-        gradient_input = matrix_cast<float>(gaussian_randm(5,5, 2));
+        const long nr = 3;
+        const long nc = 3;
+        resizable_tensor src(5,5,nr,nr), dest(5,5,nr,nc), gradient_input(5,5,nr,nc);
+        src = matrix_cast<float>(gaussian_randm(5,5*nr*nc, 0));
+        dest = matrix_cast<float>(gaussian_randm(5,5*nr*nc, 1));
+        gradient_input = matrix_cast<float>(gaussian_randm(5,5*nr*nc, 2));