From 4155592aa70588c6ce545efa8d3459e90e5134e1 Mon Sep 17 00:00:00 2001
From: liyin <liyin@xiaomi.com>
Date: Fri, 8 Mar 2019 13:18:55 +0800
Subject: [PATCH] Support scalar output shape

---
 mace/python/tools/visualization/index.html | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/mace/python/tools/visualization/index.html b/mace/python/tools/visualization/index.html
index f36ea260..658897fa 100644
--- a/mace/python/tools/visualization/index.html
+++ b/mace/python/tools/visualization/index.html
@@ -111,7 +111,11 @@ Click node to see details at bottom of this page.
             var output_shapes = [];
             if (typeof node["outputShape"] !== "undefined") {
                 for (var j = 0; j < node["outputShape"].length; j++) {
-                    var output_shape = node["outputShape"][j].dims.join(",");
+                    var output_shape = "";
+                    if (typeof node["outputShape"][j].dims !== "undefined") {
+                        console.log(node["outputShape"][j].dims);
+                        output_shape = node["outputShape"][j].dims.join(",");
+                    }
                     output_shapes.push(output_shape);
                 }
             }
@@ -140,11 +144,15 @@ Click node to see details at bottom of this page.
                     " min=" + node["minval"] +
                     " max=" + node["maxval"];
             }
+            var dims = "";
+            if (typeof node["dims"] != "undefined") {
+                dims = node["dims"].join(",");
+            }
             tensor_data.push({
                 "idx": tensor_data.length,
                 "name": node["name"],
                 "data_type": node["dataType"],
-                "dims": node["dims"].join(","),
+                "dims": dims,
                 "quantize_info": quantize_info
 
             })
-- 
GitLab