Fix misleading documentation in montage functions

Enet4 · web-flow · commit d58c15bbf18c · 2016-11-10T16:37:08.000Z
Tweaked the documentation of `utils.montage` and `utils.montage_batch` to inform that the input is a numpy array, not a tensor.
diff --git a/python/libs/utils.py b/python/libs/utils.py
@@ -12,8 +12,8 @@ def montage_batch(images):
 
     Parameters
     ----------
-    batch : Tensor
-        Input tensor to create montage of.
+    batch : numpy.ndarray
+        Input array to create montage of.
 
     Returns
     -------
@@ -44,8 +44,8 @@ def montage(W):
 
     Parameters
     ----------
-    W : Tensor
-        Input tensor to create montage of.
+    W : numpy.ndarray
+        Input array to create montage of.
 
     Returns
     -------
