inside my model’s forward() method i split input image into three channels, and pass each one through separate modeule. After that, i need to somehow combine each batch of color channels back together
So before merging i have three tensors with shape [batch_size,1,128,128]
After - [batch_size,3,128,128]
How do i do it?