Review comment, added docstring

apache · Oct 25, 2018 · 797b5ea · 797b5ea
1 parent bed1243
commit 797b5ea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/include/tvm/relay/type.h b/include/tvm/relay/type.h
@@ -82,7 +82,9 @@ class TensorTypeNode : public BaseTensorTypeNode {
     v->Visit("span", &span);
   }
 
-  /*! \brief Return product of elements in the shape */
+  /*! \brief Return product of elements in the shape.
+   *  \return (d1 * d_2 ... * d_n) if shape is (d_1, d_2, ..., d_n) and 1 if shape size is zero.
+   */
   TVM_DLL IndexExpr Size() const;
 
   TVM_DLL static TensorType make(Array<IndexExpr> shape, DataType dtype);