Set stringPrefix for Seq returned from split

Atry · Atry · commit b1de7136f8e2 · 2018-03-29T19:04:19.000+08:00
diff --git a/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala b/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala
@@ -940,7 +940,9 @@ trait Tensors extends OpenCL {
     def split(dimension: Int): IndexedSeq[Tensor] = {
       // TODO: override map/reduce to produce less OpenCL C code
       val newShape = shape.patch(dimension, Nil, 1)
-      new IndexedSeq[Tensor] {
+      final class TensorSeq extends IndexedSeq[Tensor] {
+
+        override def stringPrefix = "TensorSeq"
 
         val length: Int = shape(dimension)
 
@@ -973,6 +975,7 @@ trait Tensors extends OpenCL {
         }
 
       }
+      new TensorSeq
     }
 
     /**

Original file line number	Diff line number	Diff line change
`@@ -940,7 +940,9 @@ trait Tensors extends OpenCL {`
`940`	`940`	`def split(dimension: Int): IndexedSeq[Tensor] = {`
`941`	`941`	`// TODO: override map/reduce to produce less OpenCL C code`
`942`	`942`	`val newShape = shape.patch(dimension, Nil, 1)`
`943`		`- new IndexedSeq[Tensor] {`
	`943`	`+ final class TensorSeq extends IndexedSeq[Tensor] {`
	`944`	`+`
	`945`	`+ override def stringPrefix = "TensorSeq"`
`944`	`946`
`945`	`947`	`val length: Int = shape(dimension)`
`946`	`948`
`@@ -973,6 +975,7 @@ trait Tensors extends OpenCL {`
`973`	`975`	`}`
`974`	`976`
`975`	`977`	`}`
	`978`	`+ new TensorSeq`
`976`	`979`	`}`
`977`	`980`
`978`	`981`	`/**`