Lutz Roeder 5 лет назад
Родитель
Сommit
bb94cfbd08
1 измененных файлов с 5 добавлено и 0 удалено
  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) {