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.