try directly using the buffer

bioimage-io · Oct 2, 2024 · 616aa98 · 616aa98
1 parent f7e2908
commit 616aa98
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/shm/TensorBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/shm/TensorBuilder.java
@@ -142,9 +142,9 @@ private static NDArray buildFloat(SharedMemoryArray tensor, NDManager manager)
 			throw new IllegalArgumentException("Shared memory arrays must be saved in numpy format.");
 		ByteBuffer buff = tensor.getDataBufferNoHeader();
 		FloatBuffer floatBuff = buff.asFloatBuffer();
-		float[] floatArray = new float[floatBuff.capacity()];
-		floatBuff.get(floatArray);
-		NDArray ndarray = manager.create(floatBuff.array(), new Shape(ogShape));
+		//float[] floatArray = new float[floatBuff.capacity()];
+		//floatBuff.get(floatArray);
+		NDArray ndarray = manager.create(floatBuff, new Shape(ogShape));
 		return ndarray;
 	}