CoreML Catalyst MPSNNReduceFeatureChannelsArgumentMax returns all zeros

Question

I\'m using CoreML for image segmentation. I have a VNCoreMLRequest to run a model that returns a MLMultiArray. To accelerate processing the model output, I use MPSNNReduceFe