瀏覽代碼

Update mnn.js

Lutz Roeder 5 年之前
父節點
當前提交
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) {