[pycaffe] net spec layers can have ntop=0

In this case, the Function is returned instead of a Top, which can be
assigned a name if desired.

Thanks @philkr for an earlier implementation of this.

1 files changed, 3 insertions, 1 deletions

diff --git a/python/caffe/net_spec.py b/python/caffe/net_spec.py
index 16f30008..31cde7ad 100644
--- a/python/caffe/net_spec.py
+++ b/python/caffe/net_spec.py
@@ -188,7 +188,9 @@ class Layers(object):
     def __getattr__(self, name):
         def layer_fn(*args, **kwargs):
             fn = Function(name, args, kwargs)
-            if fn.ntop == 1:
+            if fn.ntop == 0:
+                return fn
+            elif fn.ntop == 1:
                 return fn.tops[0]
             else:
                 return fn.tops