From b1de7136f8e22b366c54e1ae41e11913e22929f2 Mon Sep 17 00:00:00 2001
From: Yang Bo <atryyang@thoughtworks.com>
Date: Thu, 29 Mar 2018 19:04:19 +0800
Subject: [PATCH] Set stringPrefix for Seq returned from split

---
 .../src/main/scala/com/thoughtworks/compute/Tensors.scala    | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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
     }
 
     /**