Easy way for mutil-type data input?

I’m trying joint model training,so my model’s input data is various.
example :

uid_data: [12,3,23,4,54,23,12,43,32...]  
feature_data:[1,0,2.3,23,3,0.4...]
text_data:[[1,2,5,3,23,43..],[4,34,54,34,5,...],...]
and many...

for normal way, i need to write these variables one by one to forward function ,like:

class MyModel():
    def __init__():
        pass
    def forward(self,uid_data,uid_mask,feature_data,feature_mask,text_data,text_mask):
        pass
#
uid_data=torch.LongTensor(uid)
feature_data=torch.LongTensor(feature)
text_data=torch.LongTensor(text)
...

I want to know if there is an easy way to achieve these like:

class MyModel():
    def __init__():
        pass
    def forward(self,x):
        pass

#
X_data=torch.LongTensor(train_X) # element of train_X like: [[12,3,23,4,54,23,12,43,32...] , [1,0,2.3,23,3,0.4...],[[1,2,5,3,23,43..],[4,34,54,34,5,...],...],...]

thanks a lot !

I used to concatenate them at the data loader into one X, then split them back at the beginning of “forward”