Fix misleading documentation in montage functions

Tweaked the documentation of `utils.montage` and `utils.montage_batch` to inform that the input is a numpy array, not a tensor.
yangxi49 · Nov 10, 2016 · d58c15b · d58c15b
1 parent dba8a76
commit d58c15b
Showing 1 changed file with 4 additions and 4 deletions.
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
     -------