Hi, I have a question. For the torch.nn.DataParallel, I use the following code:
model = DataParallel(MyModule1()).to('cuda:0')
y = model(x)

Does the type of x must be torch.Tensor? Can I use other types such as list, tuple …?
Thanks a lot