From 9d325ad0700f23754b54405241a2aa5f9d253219 Mon Sep 17 00:00:00 2001 From: Alex Black Date: Tue, 27 Aug 2019 23:27:41 +1000 Subject: [PATCH] Small optimization to Nd4j.readNumpy (#183) Signed-off-by: Alex Black --- .../src/main/java/org/nd4j/linalg/factory/Nd4j.java | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java index d2a4f94a4..c8baedfa5 100644 --- a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java +++ b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java @@ -2265,15 +2265,12 @@ public class Nd4j { Preconditions.checkState(data.length == numColumns, "Data has inconsistent number of columns: data length %s, numColumns %s", data.length, numColumns); data2.add(readSplit(data)); - - } - ret = Nd4j.create(dataType, data2.size(), numColumns); - for (int i = 0; i < data2.size(); i++) { - float[] row = data2.get(i); - INDArray arr = Nd4j.create(row, new long[]{1, row.length}, dataType); - ret.putRow(i, arr); + float[][] fArr = new float[data2.size()][0]; + for(int i=0; i