Solved by ptrblck in post #4 
              
              
                In that case you could write a custom model and use these two inputs directly in the forward method. 
Here is a small example: 
class MyModel(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc1 = nn.Linear(10, 10)
        self.fc2 = nn.Linear(10, 10)
        
    def forw…