diff --git a/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala b/Tensors/src/main/scala/com/thoughtworks/compute/Tensors.scala index a21a944f..1d8f6e1b 100644 --- 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 } /**