correct error creating tensors

bioimage-io · Oct 27, 2023 · 4a4a56a · 4a4a56a
1 parent e1b8bdc
commit 4a4a56a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/tensor/TensorBuilder.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/tensor/TensorBuilder.java
@@ -118,7 +118,7 @@ private static Tensor<UInt8> buildByte(
 		int[] sArr = new int[tensorShape.length];
 		for (int i = 0; i < sArr.length; i ++)
 			sArr[i] = (int) tensorShape[i];
-		blocks.copy( new long[tensorShape.length], flatArr, sArr );
+		blocks.copy( tensor.minAsLongArray(), flatArr, sArr );
 		ByteBuffer buff = ByteBuffer.wrap(flatArr);
 		Tensor<UInt8> ndarray = Tensor.create(UInt8.class, ogShape, buff);
 		return ndarray;
@@ -145,7 +145,7 @@ private static Tensor<Integer> buildInt(
 		int[] sArr = new int[tensorShape.length];
 		for (int i = 0; i < sArr.length; i ++)
 			sArr[i] = (int) tensorShape[i];
-		blocks.copy( new long[tensorShape.length], flatArr, sArr );
+		blocks.copy( tensor.minAsLongArray(), flatArr, sArr );
 		IntBuffer buff = IntBuffer.wrap(flatArr);
 		Tensor<Integer> ndarray = Tensor.create(ogShape, buff);
 		return ndarray;
@@ -172,7 +172,7 @@ private static Tensor<Float> buildFloat(
 		int[] sArr = new int[tensorShape.length];
 		for (int i = 0; i < sArr.length; i ++)
 			sArr[i] = (int) tensorShape[i];
-		blocks.copy( new long[tensorShape.length], flatArr, sArr );
+		blocks.copy( tensor.minAsLongArray(), flatArr, sArr );
 		FloatBuffer buff = FloatBuffer.wrap(flatArr);
 		Tensor<Float> ndarray = Tensor.create(ogShape, buff);
 		return ndarray;
@@ -199,7 +199,7 @@ private static Tensor<Double> buildDouble(
 		int[] sArr = new int[tensorShape.length];
 		for (int i = 0; i < sArr.length; i ++)
 			sArr[i] = (int) tensorShape[i];
-		blocks.copy( new long[tensorShape.length], flatArr, sArr );
+		blocks.copy( tensor.minAsLongArray(), flatArr, sArr );
 		DoubleBuffer buff = DoubleBuffer.wrap(flatArr);
 		Tensor<Double> ndarray = Tensor.create(ogShape, buff);
 		return ndarray;