add a toString method

apache · Oct 23, 2018 · ebc14c4 · ebc14c4
1 parent 232fd96
commit ebc14c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala b/scala-package/core/src/main/scala/org/apache/mxnet/NDArray.scala
@@ -703,6 +703,10 @@ class NDArray private[mxnet](private[mxnet] val handle: NDArrayHandle,
     checkCall(_LIB.mxNDArraySyncCopyFromCPU(handle, source, source.length))
   }
 
+  override def toString() : String = {
+    s"${this.visualize}<NDArray ${this.shape} ${this.context}>"
+  }
+
   /**
     * Visualize the internal structure of NDArray
     * @return String that show the structure