Update PredictInternational on Model.Predict.cs

DevNullx64 · Oceania2018 · commit e9f2caca5732 · 2023-05-29T13:27:22.000-05:00
Fix issue if data_handler.steps() &gt; 1
diff --git a/src/TensorFlowNET.Keras/Engine/Model.Predict.cs b/src/TensorFlowNET.Keras/Engine/Model.Predict.cs
@@ -99,7 +99,8 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)
                     }
                     else
                     {
-                        batch_outputs = tf.concat(new Tensor[] { batch_outputs, tmp_batch_outputs[0] }, axis: 0);
+                        for (int i = 0; i < batch_outputs.Length; i++)
+                            batch_outputs[i] = tf.concat(new Tensor[] { batch_outputs[i], tmp_batch_outputs[i] }, axis: 0);
                     }
 
                     var end_step = step + data_handler.StepIncrement;
@@ -116,7 +117,7 @@ Tensors run_predict_step(OwnedIterator iterator)
         {
             var data = iterator.next();
             var outputs = predict_step(data);
-            tf_with(ops.control_dependencies(new object[0]), ctl => _predict_counter.assign_add(1));
+            tf_with(ops.control_dependencies(Array.Empty<object>()), ctl => _predict_counter.assign_add(1));
             return outputs;
         }
 

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,8 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)`
`99`	`99`	`}`
`100`	`100`	`else`
`101`	`101`	`{`
`102`		`- batch_outputs = tf.concat(new Tensor[] { batch_outputs, tmp_batch_outputs[0] }, axis: 0);`
	`102`	`+ for (int i = 0; i < batch_outputs.Length; i++)`
	`103`	`+ batch_outputs[i] = tf.concat(new Tensor[] { batch_outputs[i], tmp_batch_outputs[i] }, axis: 0);`
`103`	`104`	`}`
`104`	`105`
`105`	`106`	`var end_step = step + data_handler.StepIncrement;`
`@@ -116,7 +117,7 @@ Tensors run_predict_step(OwnedIterator iterator)`
`116`	`117`	`{`
`117`	`118`	`var data = iterator.next();`
`118`	`119`	`var outputs = predict_step(data);`
`119`		`- tf_with(ops.control_dependencies(new object[0]), ctl => _predict_counter.assign_add(1));`
	`120`	`+ tf_with(ops.control_dependencies(Array.Empty<object>()), ctl => _predict_counter.assign_add(1));`
`120`	`121`	`return outputs;`
`121`	`122`	`}`
`122`	`123`