From 6297b55f28a155d9db35661731adbd473f7c0bd9 Mon Sep 17 00:00:00 2001
From: Duy-Nguyen Ta <thduynguyen@gmail.com>
Date: Wed, 30 Nov 2016 05:52:47 -0500
Subject: [PATCH] squeeze extra dims for numpy vectors

---
 cython/gtsam_utils/np_utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cython/gtsam_utils/np_utils.py b/cython/gtsam_utils/np_utils.py
index daae8f85d..84940eb8e 100644
--- a/cython/gtsam_utils/np_utils.py
+++ b/cython/gtsam_utils/np_utils.py
@@ -6,9 +6,9 @@ def Vector(*args):
     Convenient function to create numpy vector to use with gtsam cython wrapper
     Usage: Vector(1), Vector(1,2,3), Vector(3,2,4)
     """
-    ret = np.asarray(args, dtype='float')
-    while ret.ndim >= 2:
-        ret = ret[0, :]
+    ret = np.squeeze(np.asarray(args, dtype='float'))
+    if ret.ndim == 0:
+        ret = np.expand_dims(ret, axis=0)
     return ret