executor: return outputs on forward (#312)

dmlc · Nov 9, 2017 · 00d61d2 · 00d61d2
1 parent 1fc03f2
commit 00d61d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/executor.jl b/src/executor.jl
@@ -156,14 +156,16 @@ function simple_bind(self :: SymbolicNode, ctx :: Context;
 end
 
 
-function forward(self :: Executor; is_train::Bool=false, kwargs...)
+function forward(self::Executor; is_train::Bool = false, kwargs...)
   for (k,v) in kwargs
     @assert(k ∈ self.arg_dict, "Unknown argument $k")
     @assert(isa(v, NDArray), "Keyword argument $k must be an NDArray")
     copy!(self.arg_dict[k], v)
   end
 
   @mxcall(:MXExecutorForward, (MX_handle, Cint), self, is_train)
+
+  self.outputs
 end
 
 function backward(self :: Executor)