Added an implicit conversion from int--> float to support int operati…

…ons in NDArrays. (These ops were already supported in the previous versions)
apache · Jan 2, 2019 · e113a24 · e113a24
1 parent 996ead1
commit e113a24
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/MX_PRIMITIVES.scala b/scala-package/core/src/main/scala/org/apache/mxnet/MX_PRIMITIVES.scala
@@ -65,6 +65,8 @@ object MX_PRIMITIVES {
 
   implicit def MX_FloatToFloat(d: MX_FLOAT) : Float = d.data
 
+  implicit def IntToMX_Float(d: Int): MX_FLOAT = new MX_FLOAT(d.toFloat)
+
   /**
     * Mimics Double in Scala.
     * @param data