[api] alternative NDArray should not be closed in NDScope (#3490)

deepjavalibrary · Oct 2, 2024 · e623210 · e623210
1 parent b6656ed
commit e623210
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java b/api/src/main/java/ai/djl/ndarray/NDArrayAdapter.java
@@ -1346,6 +1346,7 @@ private NDArray getAlternativeArray() {
         } else {
             alternativeArray.set(getDataType().asDataType(toByteBuffer()));
         }
+        NDScope.unregister(alternativeArray);
         return alternativeArray;
     }
 }