Skip to content

Commit

Permalink
test one thing
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosuc3m committed Nov 18, 2024
1 parent 6959de8 commit fc57c88
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -118,6 +118,7 @@ private static RandomAccessibleInterval<IntType> buildFromTensorInt(Tensor<Integ

private static RandomAccessibleInterval<FloatType> buildFromTensorFloat(Tensor<Float> tensor) {
long[] arrayShape = tensor.shape();
System.out.println(Arrays.asList(arrayShape));
if (CommonUtils.int32Overflows(arrayShape, 4))
throw new IllegalArgumentException("Model output tensor with shape " + Arrays.toString(arrayShape)
+ " is too big. Max number of elements per float output tensor supported: " + Integer.MAX_VALUE / 4);
Expand Down

0 comments on commit fc57c88

Please sign in to comment.