What if we want to concate two inputs?
outputs = []
for i in range(in_channels):
for j in range(in_channels):
tensor = torch.cat(input_batch1[:,i,:,:], input_batch2[:,j,:,:], dim=1)   #transform from (64, 1, 224, 224) to (64, 32, 224, 224)
outputs.append(tensor)
result = torch.cat(outputs, dim=1)