add function sum to Symbol

liyangdev · May 10, 2016 · be7220a · be7220a
1 parent 2145e0d
commit be7220a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/scala-package/core/src/main/scala/ml/dmlc/mxnet/Symbol.scala b/scala-package/core/src/main/scala/ml/dmlc/mxnet/Symbol.scala
@@ -805,6 +805,14 @@ object Symbol {
     createFromListedSymbols("square")(Array(src))
   }
 
+  /**
+   * Take sum of the src
+   * @param src Source symbolic input to the function
+   */
+  def sum(src: Symbol): Symbol = {
+    createFromListedSymbols("sum")(Array(src))
+  }
+
   /**
    * Take sqrt of the src
    * src Source symbolic input to the function