From 23aa6b01885b47acbf04a045fb4cadf79d1c7d40 Mon Sep 17 00:00:00 2001
From: Alexander Morozov <etesial@gmail.com>
Date: Sun, 20 Mar 2016 21:23:36 +0300
Subject: [PATCH] fix/nn-config: fix NN input vector dimensions
 [SKIP_CHANGELOG]

---
 src/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 88b069f..66da78d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -168,7 +168,7 @@ fn run_mnist(model_name: Option<String>, batch_size: Option<usize>, learning_rat
     let mut confusion = ::leaf::solver::ConfusionMatrix::new(10);
     confusion.set_capacity(Some(1000));
 
-    let mut inp = SharedTensor::<f32>::new(backend.device(), &vec![batch_size, 1, 28, 28]).unwrap();
+    let mut inp = SharedTensor::<f32>::new(backend.device(), &vec![batch_size, 28, 28]).unwrap();
     let label = SharedTensor::<f32>::new(native_backend.device(), &vec![batch_size, 1]).unwrap();
     inp.add_device(native_backend.device()).unwrap();