Finished tensor_rand constructor.

c8948ee0 · Davis King · 3d1c22b9 · c8948ee0
Commit c8948ee0 authored Nov 16, 2015 by Davis King
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 4 deletions

dlib/dnn/tensor_tools.cpp dlib/dnn/tensor_tools.cpp +7 -4

No files found.
--- a/dlib/dnn/tensor_tools.cpp
+++ b/dlib/dnn/tensor_tools.cpp
@@ -4,6 +4,7 @@
 #define DLIB_TeNSOR_TOOLS_CPP_

 #include "tensor_tools.h"
+#include "../string.h"

 namespace dlib { namespace tt
 {
@@ -41,9 +42,11 @@ namespace dlib { namespace tt
    tensor_rand(
        unsigned long long seed
    ) 
-    {
-        // TODO
-    }
+#ifdef DLIB_USE_CUDA
+    :rnd(seed){}
+#else
+    {rnd.set_seed(cast_to_string(seed)); }
+#endif

    void tensor_rand::
    fill_gaussian (