Use lastAdded if add* is passed [null] or null

deeplearning4j · Mar 30, 2019 · e64184d · e64184d
1 parent faaf215
commit e64184d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...learning4j-nn/src/main/java/org/deeplearning4j/nn/conf/ComputationGraphConfiguration.java b/...learning4j-nn/src/main/java/org/deeplearning4j/nn/conf/ComputationGraphConfiguration.java
@@ -967,7 +967,7 @@ public GraphBuilder setOutputs(String... outputNames) {
          */
         public GraphBuilder addVertex(String vertexName, GraphVertex vertex, String... vertexInputs) {
 
-            if(vertexInputs.length == 0){
+            if(vertexInputs == null || (vertexInputs.length > 0 && vertexInputs[0] == null)){
                 if(lastAdded != null) {
                     vertexInputs = new String[]{lastAdded};
                 }