Can I use Switch shader that index value result goes to different AOVs ?

[stigla]

Can I use Switch shader that depending on index value, result would go to different AOVs ?

Switch shader ref: https://support.solidangle.com/display/A5AFMUG/Switch

The idea is to make different masks for each of index values.
I know I can do this with render layers. The output I'm aiming for is just a mask.
If this is wrong, please help with alternative solution.
I'm doing something more complex, but if I narrow it down to basic thing, I'm trying to output all the colors from utility shader Uniform ID as separate layers.
In other words, is it possible to pass values from custom AOV back to the shader ?