Hi, I have a ModuleDictionary where each module take similar size input, x.
I have two use cases for it:

For a single input x run all module in dictionary and create tensor by stacking all outputs.
Give a ordered map of {key, x} run each x to module in dictionary for that key and stack the outputs in that order. I can change input of this case to be batch of x and corresponding list of key.

Currently I am doing this by looping on dictionary. Is there a better way to do this.