Mapping a list of input to a module

Hi everyone,
I was reading around and couldn’t figure it out.
Is there is a way to Map multiple inputs in parallel, to a module that expects a single input?

Specifically with the following scenario in mind:


def MyCustomModule(Module):
def init():
def forward(self, single_x)
some computation
return single_output

def OtherLayer(Module):
def init():
def forward(self, x)
some computation
return multiple_x

define network:
x = OtherLayer(x)
output = map(MyCustomModule, x)

Is it possible to define such a thing where the feeding of the custom module will happen in parallel?
Or alternatively with a torch custom function?