Parcourir la source

Update mxnet.js

Lutz Roeder il y a 4 ans
Parent
commit
dcbd9e3ec2
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      source/mxnet.js

+ 2 - 2
source/mxnet.js

@@ -339,14 +339,14 @@ mxnet.Graph = class {
                 node.outputs = [];
             }
             for (const node of nodes) {
-                node.inputs = node.inputs.map((input) => {
+                node.inputs = (node.inputs || []).map((input) => {
                     return mxnet.Graph._updateOutput(nodes, input);
                 });
             }
 
             const outputCountMap = {};
             for (const node of nodes) {
-                for (const output of node.outputs) {
+                for (const output of node.outputs || []) {
                     outputCountMap[output] = (outputCountMap[output] || 0) + 1;
                 }
             }