AIAA output as float32 volume

Is it possible to receive a float 32 volume instead of a segmentation mask as the output of an auto-segmentation custom model?

I would recommend to ask this from NVidia developers in the AIAA client repository.