Get multiple outputs from MPSNNGraph

[vincent1bt]

Hello!, Is there a way to get multiple MPSNNImageNode from a MPSNNGraph?

We can use this function to specify multiple result images:

let graph = MPSNNGraph(device: commandQueue.device, resultImages: [...], resultsAreNeeded: &returnImages)

However, to run the graph the function:

graph.executeAsync(withSourceImages: [inputImage]) { (outputImage, error) in ...

Only has one MPSImage output.

And I can only get the first MPSNNImageNode from the resultImages array in the graph.executeAsync function.

Is there a way to get all the MPSNNImageNode from the resultImages array?

[hollance]

I was under the impression that this was possible these days, but I haven't used MPSNNGraph in ages, so I'm not sure how to do it.