diff options
Diffstat (limited to 'python/caffe/net_spec.py')
| -rw-r--r-- | python/caffe/net_spec.py | 4 |
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 |