to_numpy_string

Author: Oceania2018

src/TensorFlowNET.Core/Tensors/tensor_util.cs

@@ -470,7 +470,20 @@ public static Tensor shape_tensor(int[] shape)
             return ops.convert_to_tensor(shape, dtype: TF_DataType.TF_INT32, name: "shape");
         }
 
-        public static string to_numpy_string(Tensor tensor)
+        public static string to_numpy_string(Tensor array)
+        {
+            Shape shape = array.shape;
+            if (shape.ndim == 0)
+                return array[0].ToString();
+
+            var s = new StringBuilder();
+            s.Append("array(");
+            PrettyPrint(s, array);
+            s.Append(")");
+            return s.ToString();
+        }
+
+        static string Render(Tensor tensor)
         {
             if (tensor.buffer == IntPtr.Zero)
                 return "Empty";
@@ -487,7 +500,7 @@ public static string to_numpy_string(Tensor tensor)
                 else
                     return $"['{string.Join("', '", tensor.StringData().Take(25))}']";
             }
-            else if(dtype == TF_DataType.TF_VARIANT)
+            else if (dtype == TF_DataType.TF_VARIANT)
             {
                 return "<unprintable>";
             }
@@ -515,7 +528,7 @@ public static string to_numpy_string(Tensor tensor)
                 var array = tensor.ToArray<float>();
                 return DisplayArrayAsString(array, tensor.shape);
             }
-            else if(dtype == TF_DataType.TF_DOUBLE)
+            else if (dtype == TF_DataType.TF_DOUBLE)
             {
                 var array = tensor.ToArray<double>();
                 return DisplayArrayAsString(array, tensor.shape);
@@ -532,14 +545,72 @@ static string DisplayArrayAsString<T>(T[] array, Shape shape)
             if (shape.ndim == 0)
                 return array[0].ToString();
 
-            var display = "array([";
+            var display = "";
             if (array.Length < 10)
                 display += string.Join(", ", array);
             else
-                display += string.Join(", ", array.Take(3)) + " ... " + string.Join(", ", array.Skip(array.Length - 3));
-            return display + "])";
+                display += string.Join(", ", array.Take(3)) + ", ..., " + string.Join(", ", array.Skip(array.Length - 3));
+            return display;
+        }
+
+        static void PrettyPrint(StringBuilder s, Tensor array, bool flat = false)
+        {
+            var shape = array.shape;
+
+            if (shape.Length == 1)
+            {
+                s.Append("[");
+                s.Append(Render(array));
+                s.Append("]");
+                return;
+            }
+
+            var len = shape[0];
+            s.Append("[");
+
+            if (len <= 10)
+            {
+                for (int i = 0; i < len; i++)
+                {
+                    PrettyPrint(s, array[i], flat);
+                    if (i < len - 1)
+                    {
+                        s.Append(", ");
+                        if (!flat)
+                            s.AppendLine();
+                    }
+                }
+            }
+            else
+            {
+                for (int i = 0; i < 5; i++)
+                {
+                    PrettyPrint(s, array[i], flat);
+                    if (i < len - 1)
+                    {
+                        s.Append(", ");
+                        if (!flat)
+                            s.AppendLine();
+                    }
+                }
+
+                s.Append(" ... ");
+                s.AppendLine();
+
+                for (int i = (int)array.size - 5; i < len; i++)
+                {
+                    PrettyPrint(s, array[i], flat);
+                    if (i < len - 1)
+                    {
+                        s.Append(", ");
+                        if (!flat)
+                            s.AppendLine();
+                    }
+                }
+            }
+
+            s.Append("]");
         }
-        
 
         public static ParsedSliceArgs ParseSlices(Slice[] slices)
         {

test/TensorFlowNET.UnitTest/Numpy/Array.Creation.Test.cs

@@ -3,6 +3,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
+using Tensorflow;
 using Tensorflow.NumPy;
 
 namespace TensorFlowNET.UnitTest.NumPy
@@ -88,5 +89,14 @@ public void meshgrid_same_ndim()
             AssetSequenceEqual(a.ToArray<int>(), new int[] { 0, 1, 2, 0, 1, 2, 0, 1, 2 });
             AssetSequenceEqual(b.ToArray<int>(), new int[] { 0, 0, 0, 1, 1, 1, 2, 2, 2 });
         }
+
+        [TestMethod]
+        public void to_numpy_string()
+        {
+            var nd = np.arange(10 * 10 * 10 * 10).reshape((10, 10, 10, 10));
+            var str = tensor_util.to_numpy_string(nd);
+            Assert.AreEqual("array([[[[0, 1, 2, ..., 7, 8, 9],", str.Substring(0, 33));
+            Assert.AreEqual("[9990, 9991, 9992, ..., 9997, 9998, 9999]]]])", str.Substring(str.Length - 45));
+        }
     }
 }