Optimize tensor copy

jignparm · jignparm · commit 4cf52d880284 · 2018-10-17T03:18:14.000Z
diff --git a/src/Microsoft.ML.OnnxTransform/OnnxUtils.cs b/src/Microsoft.ML.OnnxTransform/OnnxUtils.cs
@@ -342,11 +342,8 @@ public static void CopyTo<T>(Tensor tensor, T[] dst)
         {
             if (typeof(T) == typeof(System.Single))
             {
-                // Sonoma only takes List<T>. We need to do an extra copy to T[]
-                var listDst = new List<System.Single>();
                 var typedDst = (System.Single[])(object)dst;
-                tensor.CopyTo(listDst);
-                listDst.CopyTo(typedDst);
+                tensor.CopyTo(typedDst);
             }
             else
                 throw new NotImplementedException($"Not implemented type {typeof(T)}");

Original file line number	Diff line number	Diff line change
`@@ -342,11 +342,8 @@ public static void CopyTo<T>(Tensor tensor, T[] dst)`
`342`	`342`	`{`
`343`	`343`	`if (typeof(T) == typeof(System.Single))`
`344`	`344`	`{`
`345`		`- // Sonoma only takes List<T>. We need to do an extra copy to T[]`
`346`		`- var listDst = new List<System.Single>();`
`347`	`345`	`var typedDst = (System.Single[])(object)dst;`
`348`		`- tensor.CopyTo(listDst);`
`349`		`- listDst.CopyTo(typedDst);`
	`346`	`+ tensor.CopyTo(typedDst);`
`350`	`347`	`}`
`351`	`348`	`else`
`352`	`349`	`throw new NotImplementedException($"Not implemented type {typeof(T)}");`