fix: predict with multiple outputs

fryguy1013 · fryguy1013 · commit 93cd2b66a681 · 2023-05-10T23:22:41.000-07:00
diff --git a/src/TensorFlowNET.Keras/Engine/Model.Predict.cs b/src/TensorFlowNET.Keras/Engine/Model.Predict.cs
@@ -84,7 +84,7 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)
                 Steps = data_handler.Inferredsteps
             });
 
-            Tensor batch_outputs = null;
+            Tensors batch_outputs = null;
             _predict_counter.assign(0);
             callbacks.on_predict_begin();
             foreach (var (epoch, iterator) in data_handler.enumerate_epochs())
@@ -95,7 +95,7 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)
                     var tmp_batch_outputs = run_predict_step(iterator);
                     if (batch_outputs == null)
                     {
-                        batch_outputs = tmp_batch_outputs[0];
+                        batch_outputs = tmp_batch_outputs;
                     }
                     else
                     {

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)`
`84`	`84`	`Steps = data_handler.Inferredsteps`
`85`	`85`	`});`
`86`	`86`
`87`		`- Tensor batch_outputs = null;`
	`87`	`+ Tensors batch_outputs = null;`
`88`	`88`	`_predict_counter.assign(0);`
`89`	`89`	`callbacks.on_predict_begin();`
`90`	`90`	`foreach (var (epoch, iterator) in data_handler.enumerate_epochs())`
`@@ -95,7 +95,7 @@ Tensors PredictInternal(DataHandler data_handler, int verbose)`
`95`	`95`	`var tmp_batch_outputs = run_predict_step(iterator);`
`96`	`96`	`if (batch_outputs == null)`
`97`	`97`	`{`
`98`		`- batch_outputs = tmp_batch_outputs[0];`
	`98`	`+ batch_outputs = tmp_batch_outputs;`
`99`	`99`	`}`
`100`	`100`	`else`
`101`	`101`	`{`