close tensors i forgot to close

Author: carlosuc3m

src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java

@@ -344,9 +344,10 @@ public <T extends RealType<T> & NativeType<T>, R extends RealType<R> & NativeTyp
 		for (int i = 0; i < resultPatchTensors.size(); i++) {
 			try {
 				rais.add(ImgLib2Builder.build(resultPatchTensors.get(i)));
+				resultPatchTensors.get(i).close();
 			} catch (IllegalArgumentException ex) {
-				for (org.tensorflow.Tensor<?> tt : resultPatchTensors)
-					tt.close();
+				for (int j = i; j < resultPatchTensors.size(); j++)
+					resultPatchTensors.get(j).close();
 				throw new RunModelException(Types.stackTrace(ex));
 			}
 		}
@@ -426,6 +427,9 @@ protected List<String> inferenceFromShmas(List<String> inputs) throws IOExceptio
 			ShmBuilder.build(resultPatchTensors.get(i), name, false);
 			shmaNamesList.add(name);
 		}
+		for (org.tensorflow.Tensor<?> tt : resultPatchTensors) {
+			tt.close();
+		}
 		return shmaNamesList;
 	}