Is there a way that feeding a input to layers parallelly in PyTorch?

Question

I\'m implementing ELMo with PyTorch. I want to feed ELMo\'s CNNs, which have different filter map sizes, on word matrices but I\'m considering an efficent way. Here\'s my code:<