@nile649 I’m not exactly sure what you mean, but just remove the tensor rolling functions if you want to avoid the randomization:
rolled_tensor, h_shift, w_shift = roll_tensor(image_tensor) # randomly shift tensor to prevent tile 
unrolled_tensor, _, _ = roll_tensor(rebuilt_tensor, -h_shift, -w_shift) # Undo the above random shifts