Bladeren bron

Update mnn.js

Lutz Roeder 5 jaren geleden
bovenliggende
commit
bb94cfbd08
1 gewijzigde bestanden met toevoegingen van 5 en 0 verwijderingen
  1. 5 0
      source/mnn.js

+ 5 - 0
source/mnn.js

@@ -65,6 +65,11 @@ mnn.Graph = class {
         this._inputs = [];
         this._outputs = [];
         const inputSet = new Set();
+        for (let i = 0; i < net.tensorName.length; i++) {
+            if (net.tensorName[i] === '') {
+                net.tensorName[i] = '\n' + i.toString();
+            }
+        }
         for (let i = 0; i < net.oplists.length; i++) {
             const op = net.oplists[i];
             if (op.type === mnn.schema.OpType.Input) {