Update node.rs

zkonduit · Dec 21, 2023 · 54b5542 · 54b5542
1 parent 60dadb0
commit 54b5542
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/graph/node.rs b/src/graph/node.rs
@@ -630,15 +630,14 @@ impl Node {
         out_scale = opkind.out_scale(in_scales)?;
 
         // get the output shape
-        let mut out_dims = node_output_shapes(&node, symbol_values)?;
+        let out_dims = node_output_shapes(&node, symbol_values)?;
+        // nodes vs subgraphs always have a single output
+        let mut out_dims = out_dims[0].clone();
 
         if out_dims.is_empty() {
-            out_dims = vec![vec![1]];
+            out_dims = vec![1];
         }
 
-        // nodes vs subgraphs always have a single output
-        let out_dims = out_dims[0].clone();
-
         Ok(Node {
             idx,
             opkind,