My concern is the copy. Let’s say my images are as below where each letter represents a pixel,
{|abc|, |def|, |ghi|, |jkl|} -> NET -> IMAGE_JIGSAW_CONCAT -->

|abc|def|
|ghi|jkl|

How do I perform what I am calling IMAGE_JIGSAW_CONCAT to shuffle and concatenate the outputs so that pixel a in the output backpropagates to pixel a in the input, and so on?
Will just using torch.cat do the trick?