Fix shape inference example

apache · Sep 13, 2017 · 7ab7d45 · 7ab7d45
1 parent c277e11
commit 7ab7d45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/how_to/add_op_in_backend.md b/docs/how_to/add_op_in_backend.md
@@ -141,7 +141,7 @@ inline bool QuadraticOpShape(const nnvm::NodeAttrs& attrs,
 
   SHAPE_ASSIGN_CHECK(*out_attrs, 0, in_attrs->at(0));
   SHAPE_ASSIGN_CHECK(*in_attrs, 0, out_attrs->at(0));
-  return out_attrs->at(0).Size() != 0U;
+  return out_attrs->at(0).ndim() != 0U && out_attrs->at(0).Size() != 0U;
 }
 ```
 Here are a few things to note about the above function: