Merge pull request #105 from Atry/Tensor.scalar

Atry · web-flow · commit 4eed067a4a59 · 2018-03-26T10:09:45.000+08:00
Add the scalar method
diff --git a/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala b/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala
@@ -438,7 +438,11 @@ trait Tensors extends OpenCL {
       }
     }
 
-    def fill(value: Float, shape0: Array[Int], padding: Float = 0.0f) = {
+    def scalar(value: Float, padding: Float = 0.0f): Tensor = {
+      fill(value, Tensors.ScalarShape, padding)
+    }
+
+    def fill(value: Float, shape0: Array[Int], padding: Float = 0.0f): Tensor = {
       val padding0 = padding
       new {
         val padding: Float = padding0

Original file line number	Diff line number	Diff line change
`@@ -438,7 +438,11 @@ trait Tensors extends OpenCL {`
`438`	`438`	`}`
`439`	`439`	`}`
`440`	`440`
`441`		`- def fill(value: Float, shape0: Array[Int], padding: Float = 0.0f) = {`
	`441`	`+ def scalar(value: Float, padding: Float = 0.0f): Tensor = {`
	`442`	`+ fill(value, Tensors.ScalarShape, padding)`
	`443`	`+ }`
	`444`	`+`
	`445`	`+ def fill(value: Float, shape0: Array[Int], padding: Float = 0.0f): Tensor = {`
`442`	`446`	`val padding0 = padding`
`443`	`447`	`new {`
`444`	`448`	`val padding: Float = padding0`