don't squeeze blob arrays for python
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Thu, 15 May 2014 03:17:09 +0000 (20:17 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Thu, 15 May 2014 03:17:16 +0000 (20:17 -0700)
commit8af33e8ca2f56a3bef23935990b1c9ed65629918
tree8282a8958174b0a5f9867b3b015d02f43136b619
parentaf0b857d5481cf86ff60c59e3eab71d1175ce8c6
don't squeeze blob arrays for python

Preserve the non-batch dimensions of blob arrays, even for singletons.

The forward() and backward() helpers take lists of ndarrays instead of a
single ndarray per blob, and lists of ndarrays are likewise returned.

Note that for output the blob array could actually be returned as a
single ndarray instead of a list.
python/caffe/pycaffe.py