fix assertion

piggy2008 · Nov 26, 2013 · 1e4e950 · 1e4e950
1 parent 4ae73b1
commit 1e4e950
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DBN/rbmtrain.m b/DBN/rbmtrain.m
@@ -1,6 +1,6 @@
 function rbm = rbmtrain(rbm, x, opts)
     assert(isfloat(x), 'x must be a float');
-    assert(all(x>=0) && all(x<=1), 'all data in x must be in [0:1]');
+    assert(all(x(:)>=0) && all(x(:)<=1), 'all data in x must be in [0:1]');
     m = size(x, 1);
     numbatches = m / opts.batchsize;