Optimize tensor copy (#1275)

dotnet · Oct 18, 2018 · c6d4e62 · c6d4e62
1 parent a44e131
commit c6d4e62
Showing 1 changed file with 1 addition and 4 deletions.
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)}");